mixup.pytorch
An implementation of mixup: Beyond Empirical Risk Minimization by Hongyi Zhang, Moustapha Cisse, Yann N. Dauphin, and David Lopez-Paz.
python main.py [--batch_size 128 --epochs 400 --steps 250 300 --dataset cifar10 --model resnet20 --alpha 1 [--baseline]]
Requirements
Results
CIFAR10 on ResNet-20
| EMP | mixup |
---|
max. test accuracy | 92% | 93% |