greenfantasy / fast-adversarial-training Goto Github PK
View Code? Open in Web Editor NEWTraining robust models takes a looong time since generating adversaries is so expensive. We design a parallel algorithm for training large robust models using PyTorch C++/MPI and show it runs fast!