search

Megum1 / DFST

[AAAI'21] Deep Feature Space Trojan Attack of Neural Networks by Controlled Detoxification
https://www.cs.purdue.edu/homes/cheng535/
MIT License
28 stars 2 forks source link

Where is NiN? #1

Closed CHR-ray closed 2 years ago

CHR-ray commented 2 years ago

I find the this line raise an error: nin = NiN(num_classes) So, where is your NiN block?

Megum1 commented 2 years ago

We have fixed the problem. Our current version only supports VGG. You can customize your own network structure in "train.py" and generate your attacked model.