update default training batch for PartA2 and training epochs for centerpoint

3365a5e8 · Shaoshuai Shi · 82987c4d · 3365a5e8 · 3365a5e8
Commit 3365a5e8 authored Nov 24, 2021 by Shaoshuai Shi
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tools/cfgs/waymo_models/PartA2.yaml tools/cfgs/waymo_models/PartA2.yaml +1 -1

tools/cfgs/waymo_models/centerpoint.yaml tools/cfgs/waymo_models/centerpoint.yaml +1 -1

No files found.
--- 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

--- 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