YOPO_reproduce

Reproduce works of "You Only Propagate Once: Accelerating Adversarial Training via Maximal Principle" .

Prerequisites

• PyTorch 1.0.1
• Python 3
• easydict

How to run the code

PGD training:

python MNIST/train_pgd.py

YOPO training

python MNIST/train.py

Cite

@article{zhang2019you,
  title={You Only Propagate Once: Accelerating Adversarial Training via Maximal Principle},
  author={Zhang, Dinghuai and Zhang, Tianyuan and Lu, Yiping and Zhu, Zhanxing and Dong, Bin},
  journal={arXiv preprint arXiv:1905.00877},
  year={2019}
}