File size: 3,427 Bytes
feef108 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 | Global:
model_name: PP-OCRv6_medium_rec
debug: false
use_gpu: true
epoch_num: 3
log_smooth_window: 20
print_batch_step: 10
save_model_dir: /workspace/bina02-ppocr/run/train-stage8-optimizer-resume-v2
save_epoch_step: 1
eval_batch_step:
- 0
- 200
cal_metric_during_train: false
pretrained_model: null
checkpoints: /workspace/bina02-ppocr/run/train-stage8-optimizer-resume-v2/resume-stage4
save_inference_dir: null
use_visualdl: false
infer_img: doc/imgs_words/ch/word_1.jpg
character_dict_path: /workspace/bina02-ppocr/train-data/manifests/persian_arabic_bina02_dict.txt
max_text_length: 192
infer_mode: false
use_space_char: true
distributed: false
d2s_train_image_shape:
- 3
- 48
- 1536
freeze_batch_norm: true
trainable_prefixes:
- backbone.blocks4
- backbone.blocks5
- head.ctc_encoder
- head.ctc_head
eval_handwriting_rows: 14998
eval_printed_rows: 8192
use_amp: false
stage8_mix_manifest: /workspace/bina02-ppocr/data/corpus-v8-optimizer-resume/stage8-mix-v1/MIX_MANIFEST.json
Optimizer:
name: Adam
beta1: 0.9
beta2: 0.999
lr:
name: Cosine
learning_rate: 3.0e-06
warmup_epoch: 0
regularizer:
name: L2
factor: 3.0e-05
Architecture:
model_type: rec
algorithm: SVTR_LCNet
Transform: null
Backbone:
name: PPLCNetV4
model_size: medium
train_pool_width: 192
force_pool_width: true
Head:
name: MultiHead
head_list:
- CTCHead:
Neck:
name: lightsvtr
dims: 192
depth: 2
mlp_ratio: 4.0
local_kernel: 7
use_guide: false
Head:
fc_decay: 1.0e-05
- NRTRHead:
nrtr_dim: 512
max_text_length: 192
Loss:
name: MultiLoss
weight_1: 1.0
weight_2: 0.0
loss_config_list:
- CTCLoss: null
- NRTRLoss: null
PostProcess:
name: CTCLabelDecode
Metric:
name: RecMetric
main_indicator: handwriting_norm_edit_dis
ignore_space: false
domain_split_index: 14998
Train:
dataset:
name: LMDBDataSet
data_dir: /workspace/bina02-ppocr/data/corpus-v8-optimizer-resume/stage8-mix-v1/train
ext_op_transform_idx: 1
transforms:
- DecodeImage:
img_mode: BGR
channel_first: false
- RecConAug:
prob: 0.0
ext_data_num: 0
image_shape:
- 48
- 1536
- 3
max_text_length: 192
- RecAug: null
- MultiLabelEncode:
gtc_encode: NRTRLabelEncode
- RecResizeImg:
image_shape:
- 3
- 48
- 1536
padding: true
- KeepKeys:
keep_keys:
- image
- label_ctc
- label_gtc
- length
- valid_ratio
loader:
shuffle: true
batch_size_per_card: 32
drop_last: false
num_workers: 4
Eval:
dataset:
name: LMDBDataSet
data_dir: /workspace/bina02-ppocr/data/corpus-v8-optimizer-resume/stage8-mix-v1/eval
transforms:
- DecodeImage:
img_mode: BGR
channel_first: false
- MultiLabelEncode:
gtc_encode: NRTRLabelEncode
- RecResizeImg:
image_shape:
- 3
- 48
- 1536
padding: true
- KeepKeys:
keep_keys:
- image
- label_ctc
- label_gtc
- length
- valid_ratio
loader:
shuffle: false
drop_last: false
batch_size_per_card: 32
num_workers: 4
profiler_options: null
|