Skip to content
Test on Mixup , effiicient.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
cifar
gan
trainer
yml_test
.gitignore
LICENSE
README.md

README.md

This repo contains demo reimplementations of the CIFAR-10 training code and the GAN experiment in PyTorch based on the following paper:

Hongyi Zhang, Moustapha Cisse, Yann N. Dauphin and David Lopez-Paz. mixup: Beyond Empirical Risk Minimization. https://arxiv.org/abs/1710.09412

CIFAR-10

The following table shows the median test errors of the last 10 epochs in a 200-epoch training session. (Please refer to Section 3.2 in the paper for details.)

Model weight decay = 1e-4 weight decay = 5e-4
ERM 5.53% 5.18%
mixup 4.24% 4.68%

Generative Adversarial Networks (GAN)

Other frameworks

Acknowledgement

The CIFAR-10 reimplementation of mixup is adapted from the pytorch-cifar repository by kuangliu.

You can’t perform that action at this time.