src/ir/configs/model/tc.yaml

train:
  _target_: src.models.tc_classifier.TCClassifier
  model_name: "vinai/phobert-base"
  n_classes: 5
  lr: 1e-5
  scheduler_type: "constant_schedule_with_warmup"
  max_steps: 10000
  weight_decay: 0.01
  classifier_dropout: 0.1
  mixout: False
  freeze_encoder: False

test:
  _target_: src.models.tc_classification_task.TCClassificationTask
  model: null
  function_to_apply: "sigmoid"
  return_all_scores: True