Our adversarial examples are generated using torchattacks.

import torchattacks

atk = torchattacks.PGD(model, eps=8/255, alpha=2/255, steps=4)

# Save
atk.save(data_loader, save_path="./AE.pt", verbose=True)
  
# Load
adv_loader = atk.load(load_path="./AE.pt")