File size: 218 Bytes
25fc230 |
1 2 3 4 5 6 7 8 9 10 11 12 |
name: roberta_text_classification
config_type: train
device: cuda
init_weights_from: hezarai/roberta-base-fa
seed: 42
batch_size: 8
metrics:
f1:
task: multiclass
num_train_epochs: 5
checkpoints_dir: checkpoints/
|