| config_type: trainer | |
| device: cuda | |
| init_weights_from: hezarai/bert-base-fa | |
| num_dataloader_workers: 0 | |
| seed: 42 | |
| optimizer: | |
| name: adam | |
| config_type: OPTIMIZER | |
| lr: 2.0e-05 | |
| weight_decay: 0.0 | |
| scheduler: | |
| name: reduce_on_plateau | |
| config_type: LR_SCHEDULER | |
| verbose: true | |
| batch_size: 8 | |
| use_amp: false | |
| metrics: | |
| - seqeval | |
| num_epochs: 5 | |
| save_freq: 1 | |
| checkpoints_dir: checkpoints/ | |