search

tingxueronghua / pytorch-classification-advprop

MIT License
105 stars 16 forks source link

Where is the Auxiliary BN in your codes? #5

Closed yangjiao-tj closed 3 years ago

yangjiao-tj commented 3 years ago

I have read the AdvResNet in your net.py, but I can not find anything about the Auxiliary BN, which is an essential part in this paper you reproduced.

tingxueronghua commented 3 years ago

Sorry for replying late! I am busy with my undergraduate thesis project these days. Yes, in net.py, we don't specify the normalization layer, and we send MixBatchNorm2d in imagenet.py into the model, which contains main and auxiliary batch normalization layers. You could check ResNet in net.py, and could notice that the normalization layers are given by the variable "norm_layer", which could be specified outside. If you still have any other question, you could send emails to hanyc17@mails.tsinghua.edu.cn, which I should reply much faster.