rec_mv3_none_none_ctc.yml 918 字节
Newer Older
L
LDOUBLEV 已提交
1 2 3 4 5 6
Global:
  algorithm: Rosetta
  use_gpu: true
  epoch_num: 72
  log_smooth_window: 20
  print_batch_step: 10
L
LDOUBLEV 已提交
7
  save_model_dir: output/rec_Rosetta
L
LDOUBLEV 已提交
8 9 10 11
  save_epoch_step: 3
  eval_batch_step: 2000
  train_batch_size_per_card: 256
  test_batch_size_per_card: 256
T
tink2123 已提交
12
  drop_last: false
L
LDOUBLEV 已提交
13 14 15 16 17 18
  image_shape: [3, 32, 100]
  max_text_length: 25
  character_type: en
  loss_type: ctc
  reader_yml: ./configs/rec/rec_benchmark_reader.yml
  pretrain_weights: 
19 20
  checkpoints:
  save_inference_dir:
T
tink2123 已提交
21
  infer_img:
22

L
LDOUBLEV 已提交
23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
Architecture:
  function: ppocr.modeling.architectures.rec_model,RecModel

Backbone:
  function: ppocr.modeling.backbones.rec_mobilenet_v3,MobileNetV3
  scale: 0.5
  model_name: large

Head:
  function: ppocr.modeling.heads.rec_ctc_head,CTCPredict
  encoder_type: reshape
  
Loss:
  function: ppocr.modeling.losses.rec_ctc_loss,CTCLoss

Optimizer:
  function: ppocr.optimizer,AdamDecay
  base_lr: 0.001
  beta1: 0.9
  beta2: 0.999