Skip to content

Commit

Permalink
Added new configs for w size
Browse files Browse the repository at this point in the history
  • Loading branch information
kylevedder committed May 3, 2024
1 parent 752684a commit b250faa
Show file tree
Hide file tree
Showing 6 changed files with 33 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
_base_ = "./val_debug.py"

model = dict(args=dict(lr=0.000008))
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
_base_ = "./val_debug.py"

model = dict(
args=dict(
minibatch_size=5,
lr=0.008,
)
)
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
_base_ = "./val_debug_w5_lr_em3.py"

model = dict(
args=dict(
lr=0.0008,
)
)
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
_base_ = "./val_debug_w5_lr_em3.py"

model = dict(
args=dict(
lr=0.00008,
)
)
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
_base_ = "./val_debug_w5_lr_em3.py"

model = dict(
args=dict(
lr=0.000008,
)
)
2 changes: 1 addition & 1 deletion docker/Dockerfile_cpuonly
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ RUN pip install -e setup

RUN pip install FastGeodis==1.0.5 --no-build-isolation

RUN pip install bucketed-scene-flow-eval==2.0.6
RUN pip install bucketed-scene-flow-eval==2.0.8


ENV PYTHONPATH=/project:/bucketed_scene_flow_eval:/:${PYTHONPATH}
Expand Down

0 comments on commit b250faa

Please sign in to comment.