config.json

{
  "n_classes": 4,
  "n_training_steps": 4485,
  "n_warmup_steps": 897
}