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