Trustworthy-AI-Group/TransferAttack

about the cuda memory

Truman168 opened this issue · 2 comments

Truman168 commented

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

xiaosen-wang commented

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

Truman168 commented

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

thanks very much！！！