kaggle cifar10

kaggle cifar10 resnet18 baseline

git clone https://github.com/L1aoXingyu/kaggle-cifar10.git

data4data

sudo apt-get install p7zip
cd data;
p7zip -d train.7z;
p7zip -d test.7z;
cd ..; python3 preprocess.py;

baseline

python3 train.py --bs=128  # cpu 
python3 train.py --bs=128 --use_gpu # gpu 

baseline, bs batch size, use_gpu gpu train.py``checkpoints``checkpoints

baseline checkpoints load

python3 submission.py --model_path='checkpoints/model_best.pth.tar' --use_gpu

submission.csv kaggle