File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 1313lr = 7e-5
1414train_batch_size_per_gpu = 2
1515train_num_workers = 8
16- test_batch_size_per_gpu = 1 # can't batch inference now
16+ test_batch_size_per_gpu = 1 # can't batch infer now
1717test_num_workers = 8
1818only_label_first_subword = True # select label process strategy
1919# ====================================================================
8989 pin_memory = True ,
9090 persistent_workers = True ,
9191 sampler = dict (type = 'InfiniteSampler' , shuffle = True ),
92- collate_fn = dict (type = 'long_text_data_collate ' , training = True ),
92+ collate_fn = dict (type = 'ser_collate ' , training = True ),
9393 dataset = train_dataset )
9494val_dataloader = dict (
9595 batch_size = test_batch_size_per_gpu ,
9696 num_workers = test_num_workers ,
9797 pin_memory = True ,
9898 persistent_workers = True ,
9999 sampler = dict (type = 'DefaultSampler' , shuffle = False ),
100- collate_fn = dict (type = 'long_text_data_collate ' , training = False ),
100+ collate_fn = dict (type = 'ser_collate ' , training = False ),
101101 dataset = test_dataset )
102102test_dataloader = val_dataloader
103103# ====================================================================
Original file line number Diff line number Diff line change 1- config=' /Users/wangnu/Documents/GitHub/mmocr/projects/LayoutLMv3/configs/ser/layoutlmv3_1k_xfund_zh_1gpu_bs8 .py'
1+ config=' /Users/wangnu/Documents/GitHub/mmocr/projects/LayoutLMv3/configs/ser/layoutlmv3_1k_xfund_zh_1xbs8 .py'
22
33export TOKENIZERS_PARALLELISM=false
44export OMP_NUM_THREADS=1
You can’t perform that action at this time.
0 commit comments