The main body of the code is from https://github.com/MadryLab/mnist_challenge, where the usage of the code is listed.
My contribution is modifying pgd_attack.py to test momentum iterative FGSM, Nesterov momentum FDM and momentum iterative FGSM with linear search.