albertmillan / adversarial-training-pytorch Goto Github PK
View Code? Open in Web Editor NEWImplementation of adversarial training under fast-gradient sign method (FGSM), projected gradient descent (PGD) and CW using Wide-ResNet-28-10 on cifar-10. Sample code is re-usable despite changing the model or dataset.
License: MIT License