update README

README.md

@@ -110,7 +110,7 @@ All models are trained with 8 GTX 1080Ti GPUs and are available for download.
 
 ### Waymo Open Dataset Baselines
 We provide the setting of `DATA_CONFIG.SAMPLED_INTERVAL` on the Waymo Open Dataset (WOD) to subsample partial samples for training and evaluation, 
-so you could also play with WOD by setting a smaller `DATA_CONFIG.SAMPLED_INTERVAL` if you only have limited GPU resources. 
+so you could also play with WOD by setting a smaller `DATA_CONFIG.SAMPLED_INTERVAL` even if you only have limited GPU resources. 
 
 By default, all models are trained with **20% data (~32k frames)** of all the training samples on 8 GTX 1080Ti GPUs, and the results of each cell here are mAP/mAPH calculated by the official Waymo evaluation metrics on the **whole** validation set.    
 

tools/cfgs/waymo_models/PartA2.yaml

@@ -170,8 +170,8 @@ MODEL:
 
 
 OPTIMIZATION:
-    BATCH_SIZE_PER_GPU: 4
-    NUM_EPOCHS: 80
+    BATCH_SIZE_PER_GPU: 3
+    NUM_EPOCHS: 30
 
     OPTIMIZER: adam_onecycle
     LR: 0.01

tools/cfgs/waymo_models/pv_rcnn.yaml

@@ -203,7 +203,7 @@ MODEL:
 
 
 OPTIMIZATION:
-    BATCH_SIZE_PER_GPU: 4
+    BATCH_SIZE_PER_GPU: 2
     NUM_EPOCHS: 30
 
     OPTIMIZER: adam_onecycle

tools/cfgs/waymo_models/second.yaml

@@ -100,7 +100,7 @@ MODEL:
 
 
 OPTIMIZATION:
-    BATCH_SIZE: 4
+    BATCH_SIZE_PER_GPU: 4
     NUM_EPOCHS: 30
 
     OPTIMIZER: adam_onecycle