hyperparameters.json

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