File size: 214 Bytes
d57bcf7 |
1 2 3 4 5 6 7 8 9 10 11 12 |
name: bert_text_classification
config_type: train
device: cuda
init_weights_from: hezar-ai/bert-base-fa
seed: 42
batch_size: 8
metrics:
f1:
task: multiclass
num_train_epochs: 10
checkpoints_dir: checkpoints/
|