config_training_test.json

{
  "mode": "train",
  "model_type": "distilbert",
  "model_name": "distilbert-base-uncased",
  "batch_size": 1,
  "num_epochs": 1,
  "initial_learning_rate": 2e-5
}