pointpillars_hv_secfpn_sbn-all_16xb2-2x_waymo-3d-3class.py 521 Bytes
Newer Older
Wenwei Zhang's avatar
Wenwei Zhang committed
1
_base_ = [
2
    '../_base_/models/pointpillars_hv_secfpn_waymo.py',
Wenwei Zhang's avatar
Wenwei Zhang committed
3
    '../_base_/datasets/waymoD5-3d-3class.py',
4
    '../_base_/schedules/schedule-2x.py',
Wenwei Zhang's avatar
Wenwei Zhang committed
5
6
7
8
    '../_base_/default_runtime.py',
]

# data settings
9
train_dataloader = dict(dataset=dict(dataset=dict(load_interval=1)))
10
11
12
13
14
# Default setting for scaling LR automatically
#   - `enable` means enable scaling LR automatically
#       or not by default.
#   - `base_batch_size` = (16 GPUs) x (2 samples per GPU).
auto_scale_lr = dict(enable=False, base_batch_size=32)