This implements one of result networks from Large-scale evolution of image classifiers by Esteban Real, et. al.
- Install pytorch (I recommend anaconda environment.)
- Install scikit-learn
- Download CIFAR10 dataset
- Copy two files to
{torchvision_path}/models
cp {__init__.py,evloution.py} {torchvision_path}/models
- Run
python main.py -a evolution {cifar10_data_dir}
cf. How to know {torchvision_path}?
import torchvision
print(torchvision.__file__)
-
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.
-
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. -
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.