Create configs/training_config.yaml

configs/training_config.yaml

@@ -0,0 +1,9 @@
+model_name: bert-base-uncased
+num_train_epochs: 2
+per_device_train_batch_size: 16
+per_device_eval_batch_size: 32
+output_dir: ./outputs
+logging_dir: ./logs
+evaluation_strategy: epoch
+logging_steps: 20
+report_to: mlflow