resnet18_cifar10 / hyperparameters.json
edadaltocg's picture
Upload hyperparameters.json
aa54e66
raw
history blame
547 Bytes
{
"config": "scripts/train_configs/cifar10.json",
"model": "resnet18_cifar10",
"dataset": "cifar10",
"batch_size": 128,
"epochs": 300,
"validation_frequency": 5,
"seed": 1,
"criterion": "CrossEntropyLoss",
"criterion_kwargs": {},
"optimizer": "SGD",
"lr": 0.1,
"optimizer_kwargs": {
"momentum": 0.9,
"weight_decay": 0.0005,
"nesterov": "True"
},
"scheduler": "ReduceLROnPlateau",
"scheduler_kwargs": {
"factor": 0.1,
"patience": 3,
"threshold": 0.001,
"mode": "max"
},
"debug": false
}