diff --git a/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt-openset.py b/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt-openset.py index 71666193..bc3d52a1 100644 --- 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) diff --git a/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py b/configs/kie/sdmgr/sdmgr_novisual_60e_wildreceipt.py index 6f979e91..b56c2b9b 100644 --- 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) diff --git a/configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py b/configs/kie/sdmgr/sdmgr_unet16_60e_wildreceipt.py index 030f3b2c..d49cbbc3 100644 --- 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) diff --git a/configs/textdet/dbnet/dbnet_resnet18_fpnc_100k_synthtext.py b/configs/textdet/dbnet/dbnet_resnet18_fpnc_100k_synthtext.py index dba5fd96..c992475c 100644 --- 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) diff --git a/configs/textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py b/configs/textdet/dbnet/dbnet_resnet18_fpnc_1200e_icdar2015.py index 5294552d..13751a4a 100644 --- 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) diff --git a/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_100k_synthtext.py b/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_100k_synthtext.py index 63919808..19c94f89 100644 --- 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) diff --git a/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_1200e_icdar2015.py b/configs/textdet/dbnet/dbnet_resnet50-dcnv2_fpnc_1200e_icdar2015.py index ab05a2f2..074cf74b 100644 --- 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) diff --git a/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_100k_synthtext.py b/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_100k_synthtext.py index 6a12fb54..078cb958 100644 --- 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) diff --git a/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py b/configs/textdet/dbnetpp/dbnetpp_resnet50-dcnv2_fpnc_1200e_icdar2015.py index be14e04f..6fe19265 100644 --- 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) diff --git a/configs/textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py b/configs/textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py index 6f876ce8..c6a42b07 100644 --- 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) diff --git a/configs/textdet/fcenet/fcenet_resnet50-dcnv2_fpn_1500e_ctw1500.py b/configs/textdet/fcenet/fcenet_resnet50-dcnv2_fpn_1500e_ctw1500.py index 9e61f883..c08bb16e 100644 --- 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) diff --git a/configs/textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py b/configs/textdet/fcenet/fcenet_resnet50_fpn_1500e_icdar2015.py index 93d332d0..5ad6fab3 100644 --- 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) diff --git a/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_ctw1500.py b/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_ctw1500.py index 5c269aa2..fb018655 100644 --- 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) diff --git a/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_icdar2015.py b/configs/textdet/maskrcnn/mask-rcnn_resnet50_fpn_160e_icdar2015.py index 07ff1426..399619c9 100644 --- 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) diff --git a/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_ctw1500.py b/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_ctw1500.py index d7142ddc..166b4b14 100644 --- 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) diff --git a/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_icdar2015.py b/configs/textdet/panet/panet_resnet18_fpem-ffm_600e_icdar2015.py index efeb070d..4a03cb2d 100644 --- 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) diff --git a/configs/textdet/panet/panet_resnet50_fpem-ffm_600e_icdar2017.py b/configs/textdet/panet/panet_resnet50_fpem-ffm_600e_icdar2017.py index 489aa154..ba8d37c4 100644 --- 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) diff --git a/configs/textdet/psenet/psenet_resnet50_fpnf_600e_ctw1500.py b/configs/textdet/psenet/psenet_resnet50_fpnf_600e_ctw1500.py index 7fa4eb29..9f36af2c 100644 --- 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) diff --git a/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2015.py b/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2015.py index 11d7ecf8..fc556178 100644 --- 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) diff --git a/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2017.py b/configs/textdet/psenet/psenet_resnet50_fpnf_600e_icdar2017.py index ad472a21..a813ea08 100644 --- 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) diff --git a/configs/textdet/textsnake/textsnake_resnet50_fpn-unet_1200e_ctw1500.py b/configs/textdet/textsnake/textsnake_resnet50_fpn-unet_1200e_ctw1500.py index 484b4f26..525c397f 100644 --- 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) diff --git a/configs/textrecog/abinet/abinet-vision_20e_st-an_mj.py b/configs/textrecog/abinet/abinet-vision_20e_st-an_mj.py index b6f220b8..39a60f78 100644 --- 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) diff --git a/configs/textrecog/abinet/abinet_20e_st-an_mj.py b/configs/textrecog/abinet/abinet_20e_st-an_mj.py index 078bebf4..85b00cd9 100644 --- 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) diff --git a/configs/textrecog/crnn/crnn_mini-vgg_5e_mj.py b/configs/textrecog/crnn/crnn_mini-vgg_5e_mj.py index 7fd16506..acc76cdd 100644 --- 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) diff --git a/configs/textrecog/master/master_resnet31_12e_st_mj_sa.py b/configs/textrecog/master/master_resnet31_12e_st_mj_sa.py index 214b2db5..4695e4cf 100644 --- 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) diff --git a/configs/textrecog/nrtr/nrtr_modality-transform_6e_st_mj.py b/configs/textrecog/nrtr/nrtr_modality-transform_6e_st_mj.py index 452831ed..89784a0e 100644 --- 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) diff --git a/configs/textrecog/nrtr/nrtr_resnet31-1by16-1by8_6e_st_mj.py b/configs/textrecog/nrtr/nrtr_resnet31-1by16-1by8_6e_st_mj.py index f82980ae..3cc9a0d3 100644 --- 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) diff --git 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 index 5438cef9..2a9edbf1 100644 --- 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) diff --git 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 index 96626e48..cfcdf502 100644 --- 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) diff --git a/configs/textrecog/satrn/satrn_shallow_5e_st_mj.py b/configs/textrecog/satrn/satrn_shallow_5e_st_mj.py index 76b64758..16a7ef50 100644 --- 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)