提交 fb9b01a1 编写于 作者: 文幕地方's avatar 文幕地方

update tablemv3 config

上级 f2c986d6
...@@ -17,7 +17,7 @@ Global: ...@@ -17,7 +17,7 @@ Global:
# for data or label process # for data or label process
character_dict_path: ppocr/utils/dict/table_structure_dict.txt character_dict_path: ppocr/utils/dict/table_structure_dict.txt
character_type: en character_type: en
max_text_length: &max_text_length 800 max_text_length: &max_text_length 500
box_format: &box_format 'xyxy' # 'xywh', 'xyxy', 'xyxyxyxy' box_format: &box_format 'xyxy' # 'xywh', 'xyxy', 'xyxyxyxy'
infer_mode: False infer_mode: False
...@@ -38,7 +38,8 @@ Architecture: ...@@ -38,7 +38,8 @@ Architecture:
Backbone: Backbone:
name: MobileNetV3 name: MobileNetV3
scale: 1.0 scale: 1.0
model_name: large model_name: small
disable_se: true
Head: Head:
name: TableAttentionHead name: TableAttentionHead
hidden_size: 256 hidden_size: 256
...@@ -89,7 +90,7 @@ Train: ...@@ -89,7 +90,7 @@ Train:
keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ] keep_keys: [ 'image', 'structure', 'bboxes', 'bbox_masks', 'shape' ]
loader: loader:
shuffle: True shuffle: True
batch_size_per_card: 32 batch_size_per_card: 48
drop_last: True drop_last: True
num_workers: 1 num_workers: 1
...@@ -124,5 +125,5 @@ Eval: ...@@ -124,5 +125,5 @@ Eval:
loader: loader:
shuffle: False shuffle: False
drop_last: False drop_last: False
batch_size_per_card: 16 batch_size_per_card: 48
num_workers: 1 num_workers: 1
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册