_base_ = [ | |
'../_base_/models/pointpillars_hv_secfpn_waymo.py', | |
'../_base_/datasets/waymoD5-3d-3class.py', | |
'../_base_/schedules/schedule-2x.py', | |
'../_base_/default_runtime.py', | |
] | |
# data settings | |
train_dataloader = dict(dataset=dict(dataset=dict(load_interval=1))) | |
# 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) | |