File size: 390 Bytes
83d2a3e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
config_type: trainer
device: cuda
init_weights_from: hezarai/roberta-base-fa
num_dataloader_workers: 4
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: 32
use_amp: false
metrics:
- seqeval
num_epochs: 3
save_freq: 1
checkpoints_dir: checkpoints/
|