train/train_config.yaml

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/