Harry24k / adversarial-attacks-pytorch

PyTorch implementation of adversarial attacks [torchattacks].
https://adversarial-attacks-pytorch.readthedocs.io/en/latest/index.html
MIT License
1.79k stars 337 forks source link

Add save_every_iter option #173

Open klchen0112 opened 6 months ago

klchen0112 commented 6 months ago

PR Type and Checklist

What kind of change does this PR introduce?

When saving a large number of samples, saving each batch repeatedly will consume a lot of time. I added an option to control whether generated samples are saved every epoch.