search

mnswdhw / InvGAN-Pytorch

This repository implements the Invgan defense architecture in Pytorch. InvGAN acts as an initialisation for DefenseGAN and can help in State of the Art robustness of target models against adversarial attacks.
1 stars 0 forks source link

On the training of gan networks #1

Closed ZackSoul closed 1 year ago

ZackSoul commented 1 year ago

Is there any training code about your defensegan network and Invgan network? I want to observe the specific training process

mnswdhw commented 1 year ago

Did you check these files, https://github.com/mnswdhw/InvGAN-Pytorch/blob/main/defense_gan.py and https://github.com/mnswdhw/InvGAN-Pytorch/blob/main/invgan.py These files have the training codes as well. Please note that I am using pre-trained weights of these GANs provided here https://github.com/yogeshbalaji/InvGAN These are in TensorFlow, so you will have to load them appropriately into PyTorch models.