| 12345678910111213141516171819202122232425262728293031323334353637383940 |
- Global:
- model: LaTeX_OCR_rec
- mode: check_dataset # check_dataset/train/evaluate/predict
- dataset_dir: "./dataset/ocr_rec_latexocr_dataset_example"
- device: gpu:0
- output: "output"
- CheckDataset:
- convert:
- enable: True
- src_dataset_type: PKL
- split:
- enable: False
- train_percent: null
- val_percent: null
- Train:
- epochs_iters: 20
- batch_size_train: 40
- batch_size_val: 10
- learning_rate: 0.0001
- pretrain_weight_path: null
- resume_path: null
- log_interval: 20
- eval_interval: 1
- save_interval: 1
- Evaluate:
- weight_path: "output/best_accuracy.pdparams"
- log_interval: 1
- Export:
- weight_path: https://paddle-model-ecology.bj.bcebos.com/pretrained/rec_latex_ocr_trained.pdparams
- Predict:
- model_dir: "output/best_accuracy"
- input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_formula_rec_001.png"
- kernel_option:
- run_mode: paddle
- batch_size: 1
|
