File size: 320 Bytes
6a236f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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
|