train/train_config.yaml

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/