File size: 482 Bytes
fc7e8e6 |
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/ft_cifar100.json",
"model": "vit_base_patch16_224_in21k_ft_cifar100",
"dataset": "cifar100",
"batch_size": 64,
"epochs": 10,
"validation_frequency": 1,
"seed": 1,
"criterion": "CrossEntropyLoss",
"criterion_kwargs": {},
"optimizer": "SGD",
"lr": 0.01,
"optimizer_kwargs": {
"momentum": 0.9,
"weight_decay": 0.0
},
"scheduler": "CosineAnnealingLR",
"scheduler_kwargs": {
"T_max": 10
},
"debug": false
} |