cifar.torch
cifar.torch copied to clipboard
92.45% on CIFAR-10 in Torch
cifar.torch
Newer version of this code is included in https://github.com/szagoruyko/wide-residual-networks
The code achieves 92.45% accuracy on CIFAR-10 just with horizontal reflections.
Corresponding blog post: http://torch.ch/blog/2015/07/30/cifar.html
Accuracies:
No flips | Flips |
---|---|
VGG+BN+Dropout | 91.3% |
NIN+BN+Dropout | 90.4% |
Would be nice to add other architectures, PRs are welcome!
Data preprocessing:
OMP_NUM_THREADS=2 th -i provider.lua
provider = Provider()
provider:normalize()
torch.save('provider.t7',provider)
Takes about 30 seconds and saves 1400 Mb file.
Training:
CUDA_VISIBLE_DEVICES=0 th train.lua --model vgg_bn_drop -s logs/vgg