Bina-0.2-RizehPizeh / config.yml
Reza2kn's picture
Release verified Bina 0.2 RizehPizeh 169k checkpoint
6d05289 verified
Raw
History Blame Contribute Delete
1.63 kB
Global:
model_name: PP-OCRv6_medium_rec
debug: false
use_gpu: true
epoch_num: 12
log_smooth_window: 20
print_batch_step: 20
save_model_dir: ./output
save_epoch_step: 1
eval_batch_step:
- 0
- 2640
cal_metric_during_train: false
pretrained_model: ./Bina-0.2-RizehPizeh
checkpoints: null
save_inference_dir: ./inference
use_visualdl: false
infer_img: doc/imgs_words/ch/word_1.jpg
character_dict_path: ./persian_arabic_bina02_dict.txt
max_text_length: 128
infer_mode: false
use_space_char: true
distributed: false
save_res_path: /home/rezo/bina02-ppocr/clean-restart/runs/bina02-169k-cropnorm-cw-bs16-v1/predicts.txt
d2s_train_image_shape:
- 3
- 48
- 768
Optimizer:
name: Adam
beta1: 0.9
beta2: 0.999
lr:
name: Cosine
learning_rate: 0.0003
warmup_epoch: 0
regularizer:
name: L2
factor: 0.0
Architecture:
model_type: rec
algorithm: SVTR_LCNet
Transform: null
Backbone:
name: PPLCNetV4
model_size: medium
train_pool_width: 96
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: 128
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: norm_edit_dis
ignore_space: false
profiler_options: null