Home

Awesome

PyTorch Implementation of CutMix

Usage

$ python train.py --depth 20 --use_cutmix --outdir results

Results on CIFAR-10

ModelTest Error (median of 3 runs)Training Time
WRN-20-44.561h22m
WRN-20-4, CutMix (alpha=1)3.621h22m

w/o CutMix

$ python -u train.py --depth 20 --base_channels 64 --base_lr 0.2 --scheduler cosine --seed 7 --outdir results/wo_cutmix/00

w/ CutMix

$ python -u train.py --depth 20 --base_channels 64 --base_lr 0.2 --scheduler cosine --seed 7 --use_cutmix --cutmix_alpha 1.0 --outdir results/w_cutmix/00

References