| config_type: TRAINER | |
| task: text_generation | |
| device: cuda | |
| init_weights_from: hezarai/gpt2-base-fa | |
| num_dataloader_workers: 0 | |
| seed: 42 | |
| learning_rate: 2.0e-05 | |
| weight_decay: 0.0 | |
| batch_size: 8 | |
| use_amp: false | |
| evaluate_with_generate: true | |
| metrics: | |
| - rouge | |
| num_epochs: 10 | |
| save_freq: 1 | |
| checkpoints_dir: checkpoints/ | |
| logs_dir: logs | |