| name: text_classification | |
| config_type: train | |
| device: cuda | |
| init_weights_from: hezarai/bert-base-fa | |
| num_dataloader_workers: 0 | |
| seed: 42 | |
| optimizer: | |
| lr: 2.0e-05 | |
| batch_size: 8 | |
| use_amp: false | |
| metrics: | |
| f1: | |
| task: multiclass | |
| num_epochs: 5 | |
| save_freq: 1 | |
| checkpoints_dir: checkpoints/ | |