neuralix / google_evolution

Exotic structured image classifier

This implements one of result networks from Large-scale evolution of image classifiers by Esteban Real, et. al.

Requirements

• Install pytorch (I recommend anaconda environment.)
• Install scikit-learn
• Download CIFAR10 dataset

Training

1. Copy two files to {torchvision_path}/models

cp {__init__.py,evloution.py} {torchvision_path}/models

2. Run

python main.py -a evolution {cifar10_data_dir}

cf. How to know {torchvision_path}?

import torchvision
print(torchvision.__file__)

Note

1. The numbers of channels are not in the paper and it is set by me similar with vgg. You need to adjust these for better performance.

2. If you want to adsjust learning rate on-the-fly, create lr.txt having lr value in same directory with main.py. And just change the value before some epoch you want to adopt new lr value.

3. Used CIFAR10 is image file dataset. It's not CIFAR-10 {python,Matlab,binary} version. You should convert it to conventional image files. It you don't want it, you have to use your proper dataloader.

Implemented network