search

rwightman / pytorch-nips2017-attack-example

A PyTorch baseline attack example for the NIPS 2017 adversarial competition
https://www.kaggle.com/c/nips-2017-targeted-adversarial-attack
Apache License 2.0
84 stars 27 forks source link

Do you have examples for adversarial training? #1

Closed ajing closed 6 years ago

ajing commented 6 years ago

Hi Ross,

Can I shamelessly ask whether you have any example for adversarial training?

Thanks, Jing

rwightman commented 6 years ago

I've been working on that recently, have something that's promising but it doesn't train very fast and requires multiple GPUs. I'll publish in a separate repository within the next month.

rwightman commented 6 years ago

There is an adversarial training example here: https://github.com/rwightman/pytorch-nips2017-adversarial/blob/master/python/train_adversarial_defense.py