search

Trustworthy-AI-Group / TransferAttack

TransferAttack is a pytorch framework to boost the adversarial transferability for image classification.
MIT License
268 stars 41 forks source link

about the cuda memory #74

Closed Truman168 closed 3 months ago

Truman168 commented 4 months ago

你好，我想问一下，就是我把surrogate model换成inceptionv3之后，在运行时，显存不断增加，最后直接爆掉了(总共有24G显存)，batchsize也改小了，我想问下这是怎么回事呀，是因为RAP这个方法需要很大的显存吗

xiaosen-wang commented 3 months ago

We have changed the code and you can try it again.

Truman168 commented 3 months ago

We have changed the code and you can try it again.

thanks very much！！！