Skip to content

Commit f0a03ac

Browse files
committed
规范配置文件命名
1 parent 059e203 commit f0a03ac

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

projects/LayoutLMv3/configs/ser/layoutlmv3_1k_xfund_zh_1gpu_bs8.py renamed to projects/LayoutLMv3/configs/ser/layoutlmv3_1k_xfund_zh_1xbs8.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
lr = 7e-5
1414
train_batch_size_per_gpu = 2
1515
train_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
1717
test_num_workers = 8
1818
only_label_first_subword = True # select label process strategy
1919
# ====================================================================
@@ -89,15 +89,15 @@
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)
9494
val_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)
102102
test_dataloader = val_dataloader
103103
# ====================================================================

projects/LayoutLMv3/scripts/run_ser.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
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

33
export TOKENIZERS_PARALLELISM=false
44
export OMP_NUM_THREADS=1

0 commit comments

Comments
 (0)