szagoruyko / cifar.torch

92.45% on CIFAR-10 in Torch
http://torch.ch/blog/2015/07/30/cifar.html
MIT License
174 stars 76 forks source link

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% 92.45%
NIN+BN+Dropout 90.4% 91.9%

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