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

about the defensegan pre-training model #3

Open ZackSoul opened 1 year ago

ZackSoul commented 1 year ago

Do you have the defensegan's pre-training model of pytorch? I'm trying to reproduce it, but the accuracy of my gan network reconstruction after training is very low. If you have the pre-training model, it will help me a lot. Thank you very much.

mnswdhw commented 1 year ago

I have given some conversion scripts. Try to follow them to easily obtain Pytorch compatible model weights. I have already implemented the corresponding architecture that can load those weights. I do not have the weights currently, but I think they can be obtained easily using the conversion scripts. Let me know if you have any problem following them!