config.json

{
  "model": "resnet50-pytorch",
  "dataset": "imagenet",
  "data_dir": "./imagenette2-160",
  "num_classes": 10,
  "batch_size": 128,
  "epochs": 2,
  "optimizer": "sgd",
  "scheduler": "onecycle",
  "augmentation": "strong",
  "mixup_alpha": 0.2,
  "label_smoothing": 0.1,
  "mixed_precision": true,
  "gradient_clipping": 1.0
}