cleverhans-lab / cleverhans

An adversarial example library for constructing attacks, building defenses, and benchmarking both
MIT License
6.15k stars 1.39k forks source link

Distributed Adversarial Examples Generation #1229

Open wanliAlex opened 2 years ago

wanliAlex commented 2 years ago

Hi,

The adversarial examples generation is always time-consuming in adversarial training. Is there a way to conduct multi-GPU generation in tensorflow2?

e.g. For a batch with 128 samples using 4 GPUs, can I generate 128/4 adversarial examples by each GPU at the same time and concatenate them later?

Look forward to hearing from you.