To run training LeNet-5 for CIFAR-10 classification
python main.py --config <path/to/config> --download| optimizer | dataset | architecture | config file path |
|---|---|---|---|
| Adam | CIFAR-10 | LeNet-5 | configs/cifar10/lenet_adam.json |
| K-FAC | CIFAR-10 | LeNet-5 | configs/cifar10/lenet_kfac.json |
| Noisy K-FAC | CIFAR-10 | LeNet-5 | configs/cifar10/lenet_noisykfac.json |
| VOGN | CIFAR-10 | LeNet-5 + BatchNorm | configs/cifar10/lenet_vogn.json |