| { | |
| "config": "scripts/train_configs/ft_cifar10.json", | |
| "model": "vit_base_patch16_224_in21k_ft_cifar10", | |
| "dataset": "cifar10", | |
| "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 | |
| } |