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