wrn_prime not found in adversarial robustness

[shlokk]

Hi, Thanks for the wonderful paper and open-sourcing the code. However, I had two issues in adversarial robustness code.

1) File "/ss-ood/adversarial/attacks.py", line 62, in forward logits, pen = model(adv_bx*2-1) ValueError: too many values to unpack (expected 2)

2) from models.wrn_prime import WideResNet, wrn_prime not found. I guess there is a model for which code is not present.

[Chaimmoon]

Maybe you need to modify wrn_prime to wrn?

[Sammy42779]

Hi, have you solved the first question?

[hendrycks]

return self.fc(out), out