TF2.0 port for Augmix paper - View it on GitHub