Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

MetaAttack will OOM if PyTorch>1.9 #127

Open
likuanppd opened this issue Dec 20, 2022 · 2 comments
Open

MetaAttack will OOM if PyTorch>1.9 #127

likuanppd opened this issue Dec 20, 2022 · 2 comments

Comments

@likuanppd
Copy link

likuanppd commented Dec 20, 2022

Hi jin,

we find that MetaAttack will face the OOM problem if the PyTorch version is higher than 1.9. Can you fix this problem?
For example, we attack polblogs under 20% ptb rate in one tesla-v100 (32GB), and then the cuda will OOM. We retry this in another Linux server and it reports the same error, but if we turn the pytorch version into 1.8.2, everything would be ok.

Thanks

@Leirunlin
Copy link

Leirunlin commented Feb 5, 2023

Hi!
I write an issue in #128, which seems to be helpful in avoiding OOM in MetaAttack. I didn't check the solution thoroughly on different devices. Please let me know if it could be also helpful to you.
Thanks.

@likuanppd
Copy link
Author

I've tried your solution, and it works perfectly. Good job!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants