Hi,
Thanks for your paper and code. But I got an error during running adversary = attacks.PGD(epsilon=8./255, num_steps=10, step_size=2./255).cpu() in folder adversarial.
The error was:
logits, pen = model(adv_bx * 2 - 1)ValueError: too many values to unpack (expected 2)
I don't know whether I did something wrong in the code. I just changed the .cuda() to .cpu() because I only used the CPU version.
Can anyone help me to solve this question?
Thanks so much.
Hi, Thanks for your paper and code. But I got an error during running
adversary = attacks.PGD(epsilon=8./255, num_steps=10, step_size=2./255).cpu()
in folder adversarial. The error was:logits, pen = model(adv_bx * 2 - 1)
ValueError: too many values to unpack (expected 2)
I don't know whether I did something wrong in the code. I just changed the
.cuda()
to.cpu()
because I only used the CPU version. Can anyone help me to solve this question? Thanks so much.