| { | |
| "config": "scripts/train_configs/svhn.json", | |
| "model": "densenet121_svhn", | |
| "dataset": "svhn", | |
| "batch_size": 128, | |
| "epochs": 300, | |
| "validation_frequency": 5, | |
| "seed": 1, | |
| "criterion": "CrossEntropyLoss", | |
| "criterion_kwargs": {}, | |
| "optimizer": "SGD", | |
| "lr": 0.01, | |
| "optimizer_kwargs": { | |
| "momentum": 0.9, | |
| "weight_decay": 0.0005 | |
| }, | |
| "scheduler": "MultiStepLR", | |
| "scheduler_kwargs": { | |
| "gamma": 0.1, | |
| "milestones": [ | |
| 75, | |
| 100, | |
| 150, | |
| 225 | |
| ] | |
| }, | |
| "debug": false | |
| } |