File size: 386 Bytes
8e24f26 |
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/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/
|