| { | |
| "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 | |
| } |