| name: trainer | |
| config_type: trainer | |
| output_dir: crnn-ocr-fa | |
| task: image2text | |
| device: cuda | |
| num_epochs: 8 | |
| init_weights_from: hezarai/crnn-fa-printed-96-long | |
| resume_from_checkpoint: true | |
| num_dataloader_workers: 0 | |
| dataloader_shuffle: true | |
| seed: 42 | |
| learning_rate: 2.0e-05 | |
| weight_decay: 0.0 | |
| batch_size: 16 | |
| gradient_accumulation_steps: 1 | |
| distributed: false | |
| use_cpu: false | |
| do_evaluate: true | |
| evaluate_with_generate: true | |
| metrics: | |
| - cer | |
| - wer | |
| metric_for_best_model: evaluation.cer | |
| save_enabled: true | |
| save_freq: deprecated | |
| save_steps: 7125 | |
| log_steps: 100 | |
| checkpoints_dir: checkpoints | |
| logs_dir: logs | |