File size: 463 Bytes
0fe567f 38cd970 0fe567f 38cd970 0fe567f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
{
"config": "scripts/train_configs/cifar100.json",
"model": "resnet18_cifar100",
"dataset": "cifar100",
"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
},
"scheduler": "CosineAnnealingLR",
"scheduler_kwargs": {
"T_max": 280
},
"debug": false
} |