update default training batch for PartA2 and training epochs for cent…

…erpoint
ChienLiu · Nov 24, 2021 · 3365a5e · 3365a5e
1 parent 82987c4
commit 3365a5e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tools/cfgs/waymo_models/PartA2.yaml b/tools/cfgs/waymo_models/PartA2.yaml
@@ -170,7 +170,7 @@ MODEL:
 
 
 OPTIMIZATION:
-    BATCH_SIZE_PER_GPU: 3
+    BATCH_SIZE_PER_GPU: 2
     NUM_EPOCHS: 30
 
     OPTIMIZER: adam_onecycle

diff --git a/tools/cfgs/waymo_models/centerpoint.yaml b/tools/cfgs/waymo_models/centerpoint.yaml
@@ -76,7 +76,7 @@ MODEL:
 
 OPTIMIZATION:
     BATCH_SIZE_PER_GPU: 4
-    NUM_EPOCHS: 36
+    NUM_EPOCHS: 30
 
     OPTIMIZER: adam_onecycle
     LR: 0.003