[Config] auto scale lr (#1326)

2022-09-23 14:53:48 +08:00 · 2022-09-23 14:53:48 +08:00 · 794744826e
parent c6cc37b096
commit 794744826e
30 changed files with 59 additions and 0 deletions
--- a/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt-openset.py
+++ b/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt-openset.py
@ -68,3 +68,4 @@ test_evaluator = val_evaluator

 visualizer = dict(
    type='KIELocalVisualizer', name='visualizer', is_openset=True)
+auto_scale_lr = dict(base_batch_size=4)
--- a/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py
+++ b/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py
@ -24,3 +24,5 @@ val_dataloader = dict(
    sampler=dict(type='DefaultSampler', shuffle=False),
    dataset=wildreceipt_test)
 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=4)
--- a/configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py
+++ b/configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py
@ -25,3 +25,5 @@ val_dataloader = dict(
    dataset=wildreceipt_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=4)
--- a/configs/textdet/dbnet/dbnet_resnet18_fpnc_100k_synthtext.py
+++ b/configs/textdet/dbnet/dbnet_resnet18_fpnc_100k_synthtext.py
@ -26,3 +26,5 @@ val_dataloader = dict(
    dataset=st_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py
+++ b/configs/textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py
@ -26,3 +26,5 @@ val_dataloader = dict(
    dataset=ic15_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_100k_synthtext.py
+++ b/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_100k_synthtext.py
@ -26,3 +26,5 @@ val_dataloader = dict(
    dataset=st_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_1200e_icdar2015.py
+++ b/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_1200e_icdar2015.py
@ -29,3 +29,5 @@ val_dataloader = dict(
    dataset=ic15_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_100k_synthtext.py
+++ b/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_100k_synthtext.py
@ -30,3 +30,5 @@ val_dataloader = dict(
        pipeline=_base_.test_pipeline))

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py
+++ b/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py
@ -30,3 +30,5 @@ val_dataloader = dict(
        pipeline=_base_.test_pipeline))

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py
+++ b/configs/textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py
@ -26,3 +26,5 @@ val_dataloader = dict(
    dataset=ctw_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/fcenet/fcenet_resnet50-dcnv2_fpn_1500e_ctw1500.py
+++ b/configs/textdet/fcenet/fcenet_resnet50-dcnv2_fpn_1500e_ctw1500.py
@ -54,3 +54,5 @@ val_dataloader = dict(
    dataset=ctw_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=8)
--- a/configs/textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py
+++ b/configs/textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py
@ -33,3 +33,5 @@ val_dataloader = dict(
    dataset=ic15_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=8)
--- a/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_ctw1500.py
+++ b/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_ctw1500.py
@ -55,3 +55,5 @@ val_dataloader = dict(
    dataset=ctw_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=8)
--- a/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_icdar2015.py
+++ b/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_icdar2015.py
@ -35,3 +35,5 @@ val_dataloader = dict(
    dataset=ic15_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=8)
--- a/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_ctw1500.py
+++ b/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_ctw1500.py
@ -82,3 +82,5 @@ test_dataloader = val_dataloader
 val_evaluator = dict(
    type='HmeanIOUMetric', pred_score_thrs=dict(start=0.3, stop=1, step=0.05))
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=16)
--- a/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_icdar2015.py
+++ b/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_icdar2015.py
@ -31,3 +31,5 @@ test_dataloader = val_dataloader
 val_evaluator = dict(
    type='HmeanIOUMetric', pred_score_thrs=dict(start=0.3, stop=1, step=0.05))
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=64)
--- a/configs/textdet/panet/panet_resnet50_fpem-ffm_600e_icdar2017.py
+++ b/configs/textdet/panet/panet_resnet50_fpem-ffm_600e_icdar2017.py
@ -77,3 +77,5 @@ test_dataloader = val_dataloader
 val_evaluator = dict(
    type='HmeanIOUMetric', pred_score_thrs=dict(start=0.3, stop=1, step=0.05))
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=64)
--- a/configs/textdet/psenet/psenet_resnet50_fpnf_600e_ctw1500.py
+++ b/configs/textdet/psenet/psenet_resnet50_fpnf_600e_ctw1500.py
@ -51,3 +51,5 @@ val_dataloader = dict(
    dataset=ctw_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=64 * 4)
--- a/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2015.py
+++ b/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2015.py
@ -40,3 +40,5 @@ val_dataloader = dict(
    dataset=ic15_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=64 * 4)
--- a/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2017.py
+++ b/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2017.py
@ -12,3 +12,5 @@ ic17_det_test.pipeline = _base_.test_pipeline
 train_dataloader = dict(dataset=ic17_det_train)
 val_dataloader = dict(dataset=ic17_det_test)
 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=64 * 4)
--- a/configs/textdet/textsnake/textsnake_resnet50_fpn-unet_1200e_ctw1500.py
+++ b/configs/textdet/textsnake/textsnake_resnet50_fpn-unet_1200e_ctw1500.py
@ -26,3 +26,5 @@ val_dataloader = dict(
    dataset=ctw_det_test)

 test_dataloader = val_dataloader
+
+auto_scale_lr = dict(base_batch_size=4)
--- a/configs/textrecog/abinet/abinet-vision_20e_st-an_mj.py
+++ b/configs/textrecog/abinet/abinet-vision_20e_st-an_mj.py
@ -54,3 +54,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=192 * 8)
--- a/configs/textrecog/abinet/abinet_20e_st-an_mj.py
+++ b/configs/textrecog/abinet/abinet_20e_st-an_mj.py
@ -54,3 +54,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=192 * 8)
--- a/configs/textrecog/crnn/crnn_mini-vgg_5e_mj.py
+++ b/configs/textrecog/crnn/crnn_mini-vgg_5e_mj.py
@ -45,3 +45,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=64 * 4)
--- a/configs/textrecog/master/master_resnet31_12e_st_mj_sa.py
+++ b/configs/textrecog/master/master_resnet31_12e_st_mj_sa.py
@ -55,3 +55,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=512 * 4)
--- a/configs/textrecog/nrtr/nrtr_modality-transform_6e_st_mj.py
+++ b/configs/textrecog/nrtr/nrtr_modality-transform_6e_st_mj.py
@ -51,3 +51,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=384)
--- a/configs/textrecog/nrtr/nrtr_resnet31-1by16-1by8_6e_st_mj.py
+++ b/configs/textrecog/nrtr/nrtr_resnet31-1by16-1by8_6e_st_mj.py
@ -51,3 +51,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=384)
--- a/configs/textrecog/robust_scanner/robustscanner_resnet31_5e_st-sub_mj-sub_sa_real.py
+++ b/configs/textrecog/robust_scanner/robustscanner_resnet31_5e_st-sub_mj-sub_sa_real.py
@ -64,3 +64,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=64 * 16)
--- a/configs/textrecog/sar/sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.py
+++ b/configs/textrecog/sar/sar_resnet31_parallel-decoder_5e_st-sub_mj-sub_sa_real.py
@ -63,3 +63,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=64 * 48)
--- a/configs/textrecog/satrn/satrn_shallow_5e_st_mj.py
+++ b/configs/textrecog/satrn/satrn_shallow_5e_st_mj.py
@ -47,3 +47,5 @@ val_dataloader = test_dataloader
 val_evaluator = dict(
    dataset_prefixes=['CUTE80', 'IIIT5K', 'SVT', 'SVTP', 'IC13', 'IC15'])
 test_evaluator = val_evaluator
+
+auto_scale_lr = dict(base_batch_size=64 * 8)