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.

Sign up for GitHub

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

Jump to bottom

第五章rnnlm.py显存溢出 #15

Open

akari0216 opened this issue May 19, 2022 · 2 comments

akari0216 commented May 19, 2022

如标题所示，调整了batch_size=32之后在3060的机器上运行本代码仍然会导致显存溢出，请问有无优化的方法呢？谢谢

cgnerds commented May 19, 2022

可以改成16，我修改后可以运行了。

Author

akari0216 commented May 24, 2022

可以改成16，我修改后可以运行了。

感谢，改成16后确实能跑通了。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment