Skip to content

GitLab

Commit 3365a5e8 authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

update default training batch for PartA2 and training epochs for centerpoint

parent 82987c4d
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/cfgs/waymo_models/PartA2.yaml
View file @ 3365a5e8
...@@ -170,7 +170,7 @@ MODEL: ...@@ -170,7 +170,7 @@ MODEL:
OPTIMIZATION: OPTIMIZATION:
BATCH_SIZE_PER_GPU: 3 BATCH_SIZE_PER_GPU: 2
NUM_EPOCHS: 30 NUM_EPOCHS: 30
OPTIMIZER: adam_onecycle OPTIMIZER: adam_onecycle
......
tools/cfgs/waymo_models/centerpoint.yaml
View file @ 3365a5e8
...@@ -76,7 +76,7 @@ MODEL: ...@@ -76,7 +76,7 @@ MODEL:
OPTIMIZATION: OPTIMIZATION:
BATCH_SIZE_PER_GPU: 4 BATCH_SIZE_PER_GPU: 4
NUM_EPOCHS: 36 NUM_EPOCHS: 30
OPTIMIZER: adam_onecycle OPTIMIZER: adam_onecycle
LR: 0.003 LR: 0.003
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment