checkpoint/launch-config.yml

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