diff --git a/configs/textrecog/nrtr/nrtr_modality_transform_academic.py b/configs/textrecog/nrtr/nrtr_modality_transform_academic.py index d564b24b..5c181143 100644 --- a/configs/textrecog/nrtr/nrtr_modality_transform_academic.py +++ b/configs/textrecog/nrtr/nrtr_modality_transform_academic.py @@ -6,7 +6,7 @@ _base_ = [ ] # optimizer settings -optimizer = dict(type='Adam', lr=3e-4) +optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4)) # dataset settings train_list = {{_base_.train_list}} @@ -15,7 +15,11 @@ file_client_args = dict(backend='disk') default_hooks = dict(logger=dict(type='LoggerHook', interval=50), ) train_pipeline = [ - dict(type='LoadImageFromFile', file_client_args=file_client_args), + dict( + type='LoadImageFromFile', + file_client_args=file_client_args, + ignore_empty=True, + min_size=5), dict(type='LoadOCRAnnotations', with_text=True), dict( type='RescaleToHeight', diff --git a/configs/textrecog/nrtr/nrtr_r31_1by16_1by8_academic.py b/configs/textrecog/nrtr/nrtr_r31_1by16_1by8_academic.py index d80fb96d..53a93c97 100644 --- a/configs/textrecog/nrtr/nrtr_r31_1by16_1by8_academic.py +++ b/configs/textrecog/nrtr/nrtr_r31_1by16_1by8_academic.py @@ -6,7 +6,7 @@ _base_ = [ ] # optimizer settings -optimizer = dict(type='Adam', lr=3e-4) +optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4)) # dataset settings train_list = {{_base_.train_list}} @@ -15,7 +15,11 @@ file_client_args = dict(backend='disk') default_hooks = dict(logger=dict(type='LoggerHook', interval=50)) train_pipeline = [ - dict(type='LoadImageFromFile', file_client_args=file_client_args), + dict( + type='LoadImageFromFile', + file_client_args=file_client_args, + ignore_empty=True, + min_size=5), dict(type='LoadOCRAnnotations', with_text=True), dict( type='RescaleToHeight', @@ -46,7 +50,7 @@ test_pipeline = [ train_dataloader = dict( batch_size=384, - num_workers=32, + num_workers=24, persistent_workers=True, sampler=dict(type='DefaultSampler', shuffle=True), dataset=dict( diff --git a/configs/textrecog/nrtr/nrtr_r31_1by8_1by4_academic.py b/configs/textrecog/nrtr/nrtr_r31_1by8_1by4_academic.py index bb180134..e66f2927 100644 --- a/configs/textrecog/nrtr/nrtr_r31_1by8_1by4_academic.py +++ b/configs/textrecog/nrtr/nrtr_r31_1by8_1by4_academic.py @@ -6,7 +6,7 @@ _base_ = [ ] # optimizer settings -optimizer = dict(type='Adam', lr=3e-4) +optim_wrapper = dict(type='OptimWrapper', optimizer=dict(type='Adam', lr=3e-4)) # dataset settings train_list = {{_base_.train_list}} @@ -17,7 +17,11 @@ default_hooks = dict(logger=dict(type='LoggerHook', interval=50), ) model = dict(backbone=dict(last_stage_pool=False)) train_pipeline = [ - dict(type='LoadImageFromFile', file_client_args=file_client_args), + dict( + type='LoadImageFromFile', + file_client_args=file_client_args, + ignore_empty=True, + min_size=5), dict(type='LoadOCRAnnotations', with_text=True), dict( type='RescaleToHeight',