From 3365a5e89eb5d4c7c921b4d748f205646910583a Mon Sep 17 00:00:00 2001
From: Shaoshuai Shi <shaoshuaics@gmail.com>
Date: Wed, 24 Nov 2021 17:29:13 +0100
Subject: [PATCH] update default training batch for PartA2 and training epochs
 for centerpoint

---
 tools/cfgs/waymo_models/PartA2.yaml      | 2 +-
 tools/cfgs/waymo_models/centerpoint.yaml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tools/cfgs/waymo_models/PartA2.yaml b/tools/cfgs/waymo_models/PartA2.yaml
index b029f163b..4b420f2ee 100644
--- 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
index 6fc4493ac..23850f243 100644
--- 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