Commit 327c173a authored by Shaoshuai Shi's avatar Shaoshuai Shi
Browse files

update README

parent 9d28ad44
...@@ -110,7 +110,7 @@ All models are trained with 8 GTX 1080Ti GPUs and are available for download. ...@@ -110,7 +110,7 @@ All models are trained with 8 GTX 1080Ti GPUs and are available for download.
### Waymo Open Dataset Baselines ### 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, 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. 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.
......
...@@ -170,8 +170,8 @@ MODEL: ...@@ -170,8 +170,8 @@ MODEL:
OPTIMIZATION: OPTIMIZATION:
BATCH_SIZE_PER_GPU: 4 BATCH_SIZE_PER_GPU: 3
NUM_EPOCHS: 80 NUM_EPOCHS: 30
OPTIMIZER: adam_onecycle OPTIMIZER: adam_onecycle
LR: 0.01 LR: 0.01
......
...@@ -203,7 +203,7 @@ MODEL: ...@@ -203,7 +203,7 @@ MODEL:
OPTIMIZATION: OPTIMIZATION:
BATCH_SIZE_PER_GPU: 4 BATCH_SIZE_PER_GPU: 2
NUM_EPOCHS: 30 NUM_EPOCHS: 30
OPTIMIZER: adam_onecycle OPTIMIZER: adam_onecycle
......
...@@ -100,7 +100,7 @@ MODEL: ...@@ -100,7 +100,7 @@ MODEL:
OPTIMIZATION: OPTIMIZATION:
BATCH_SIZE: 4 BATCH_SIZE_PER_GPU: 4
NUM_EPOCHS: 30 NUM_EPOCHS: 30
OPTIMIZER: adam_onecycle OPTIMIZER: adam_onecycle
......
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