[Refactor] Customize runner type (#437)

configs/3dssd/3dssd_kitti-3d-car.py

@@ -108,7 +108,7 @@ optimizer = dict(type='AdamW', lr=lr, weight_decay=0)
 optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2))
 lr_config = dict(policy='step', warmup=None, step=[80, 120])
 # runtime settings
-total_epochs = 150
+runner = dict(max_epochs=150)
 
 # yapf:disable
 log_config = dict(

configs/_base_/models/3dssd.py

@@ -84,4 +84,4 @@ optimizer = dict(type='AdamW', lr=lr, weight_decay=0)
 optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2))
 lr_config = dict(policy='step', warmup=None, step=[80, 120])
 # runtime settings
-total_epochs = 150
+runner = dict(max_epochs=150)

configs/_base_/schedules/cyclic_20e.py

@@ -21,4 +21,4 @@ momentum_config = dict(
 )
 
 # runtime settings
-total_epochs = 20
+runner = dict(type='EpochBasedRunner', max_epochs=20)

configs/_base_/schedules/cyclic_40e.py

@@ -25,7 +25,7 @@ momentum_config = dict(
     cyclic_times=1,
     step_ratio_up=0.4,
 )
-# Although the total_epochs is 40, this schedule is usually used we
-# RepeatDataset with repeat ratio N, thus the actual total epoch
+# Although the max_epochs is 40, this schedule is usually used we
+# RepeatDataset with repeat ratio N, thus the actual max epoch
 # number could be Nx40
-total_epochs = 40
+runner = dict(type='EpochBasedRunner', max_epochs=40)

configs/_base_/schedules/mmdet_schedule_1x.py

@@ -8,4 +8,4 @@ lr_config = dict(
     warmup_iters=500,
     warmup_ratio=0.001,
     step=[8, 11])
-total_epochs = 12
+runner = dict(type='EpochBasedRunner', max_epochs=12)

configs/_base_/schedules/schedule_2x.py

@@ -11,4 +11,4 @@ lr_config = dict(
     step=[20, 23])
 momentum_config = None
 # runtime settings
-total_epochs = 24
+runner = dict(type='EpochBasedRunner', max_epochs=24)

configs/_base_/schedules/schedule_3x.py

@@ -6,4 +6,4 @@ optimizer = dict(type='AdamW', lr=lr, weight_decay=0.01)
 optimizer_config = dict(grad_clip=dict(max_norm=10, norm_type=2))
 lr_config = dict(policy='step', warmup=None, step=[24, 32])
 # runtime settings
-total_epochs = 36
+runner = dict(type='EpochBasedRunner', max_epochs=36)

configs/benchmark/hv_PartA2_secfpn_4x8_cyclic_80e_pcdet_kitti-3d-3class.py

@@ -312,7 +312,7 @@ log_config = dict(
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 80
+runner = dict(max_epochs=80)
 dist_params = dict(backend='nccl', port=29506)
 log_level = 'INFO'
 find_unused_parameters = True

configs/benchmark/hv_pointpillars_secfpn_3x8_100e_det3d_kitti-3d-car.py

@@ -182,7 +182,7 @@ log_config = dict(
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 50
+runner = dict(max_epochs=50)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
 work_dir = './work_dirs/pp_secfpn_100e'

configs/benchmark/hv_pointpillars_secfpn_4x8_80e_pcdet_kitti-3d-3class.py

@@ -225,7 +225,7 @@ log_config = dict(
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 80
+runner = dict(max_epochs=80)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
 work_dir = './work_dirs/pp_secfpn_80e'

configs/benchmark/hv_second_secfpn_4x8_80e_pcdet_kitti-3d-3class.py

@@ -227,7 +227,7 @@ log_config = dict(
     ])
 # yapf:enable
 # runtime settings
-total_epochs = 80
+runner = dict(max_epochs=80)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
 work_dir = './work_dirs/sec_secfpn_80e'

configs/free_anchor/hv_pointpillars_regnet-1.6gf_fpn_sbn-all_free-anchor_strong-aug_4x8_3x_nus-3d.py

@@ -66,4 +66,4 @@ data = dict(train=dict(pipeline=train_pipeline))
 
 lr_config = dict(step=[28, 34])
 evaluation = dict(interval=36)
-total_epochs = 36
+runner = dict(max_epochs=36)

configs/free_anchor/hv_pointpillars_regnet-3.2gf_fpn_sbn-all_free-anchor_strong-aug_4x8_3x_nus-3d.py

@@ -66,4 +66,4 @@ train_pipeline = [
 data = dict(train=dict(pipeline=train_pipeline))
 lr_config = dict(step=[28, 34])
 evaluation = dict(interval=36)
-total_epochs = 36
+runner = dict(max_epochs=36)

configs/imvotenet/imvotenet_faster_rcnn_r50_fpn_2x4_sunrgbd-3d-10class.py

@@ -53,6 +53,6 @@ lr_config = dict(
     warmup_iters=500,
     warmup_ratio=0.001,
     step=[6])
-total_epochs = 8
+runner = dict(max_epochs=8)
 
 load_from = 'http://download.openmmlab.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth'  # noqa

configs/mvxnet/dv_mvx-fpn_second_secfpn_adamw_2x8_80e_kitti-3d-3class.py

@@ -248,7 +248,7 @@ log_config = dict(
 # yapf:enable
 evaluation = dict(interval=1)
 # runtime settings
-total_epochs = 40
+runner = dict(max_epochs=40)
 dist_params = dict(backend='nccl')
 log_level = 'INFO'
 work_dir = None

configs/nuimages/cascade_mask_rcnn_r50_fpn_coco-20e_20e_nuim.py

@@ -2,6 +2,6 @@ _base_ = './cascade_mask_rcnn_r50_fpn_1x_nuim.py'
 
 # learning policy
 lr_config = dict(step=[16, 19])
-total_epochs = 20
+runner = dict(max_epochs=20)
 
 load_from = 'http://download.openmmlab.com/mmdetection/v2.0/cascade_rcnn/cascade_mask_rcnn_r50_fpn_20e_coco/cascade_mask_rcnn_r50_fpn_20e_coco_bbox_mAP-0.419__segm_mAP-0.365_20200504_174711-4af8e66e.pth'  # noqa

configs/nuimages/htc_r50_fpn_coco-20e_20e_nuim.py

 _base_ = './htc_r50_fpn_coco-20e_1x_nuim.py'
 # learning policy
 lr_config = dict(step=[16, 19])
-total_epochs = 20
+runner = dict(max_epochs=20)

configs/nuimages/htc_x101_64x4d_fpn_dconv_c3-c5_coco-20e_16x1_20e_nuim.py

@@ -18,6 +18,6 @@ model = dict(
 data = dict(samples_per_gpu=1, workers_per_gpu=1)
 # learning policy
 lr_config = dict(step=[16, 19])
-total_epochs = 20
+runner = dict(max_epochs=20)
 
 load_from = 'http://download.openmmlab.com/mmdetection/v2.0/htc/htc_x101_64x4d_fpn_dconv_c3-c5_mstrain_400_1400_16x1_20e_coco/htc_x101_64x4d_fpn_dconv_c3-c5_mstrain_400_1400_16x1_20e_coco_20200312-946fd751.pth'  # noqa

configs/nuimages/mask_rcnn_r50_caffe_fpn_coco-3x_20e_nuim.py

@@ -47,6 +47,6 @@ data = dict(
 
 # learning policy
 lr_config = dict(step=[16, 19])
-total_epochs = 20
+runner = dict(max_epochs=20)
 
 load_from = 'http://download.openmmlab.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco/mask_rcnn_r50_caffe_fpn_mstrain-poly_3x_coco_bbox_mAP-0.408__segm_mAP-0.37_20200504_163245-42aa3d00.pth'  # noqa

configs/pointpillars/hv_pointpillars_secfpn_6x8_160e_kitti-3d-3class.py

@@ -83,4 +83,4 @@ evaluation = dict(interval=2)
 # PointPillars usually need longer schedule than second, we simply double
 # the training schedule. Do remind that since we use RepeatDataset and
 # repeat factor is 2, so we actually train 160 epochs.
-total_epochs = 80
+runner = dict(max_epochs=80)