training_config.json

{
  "model_name": "t5-small",
  "max_input_length": 256,
  "max_target_length": 64,
  "batch_size": 16,
  "learning_rate": 5e-05,
  "num_epochs": 1,
  "train_size": 200,
  "val_size": 40,
  "test_size": 20
}