plm-nlp-code icon indicating copy to clipboard operation
plm-nlp-code copied to clipboard

第五章rnnlm.py显存溢出

Open akari0216 opened this issue 2 years ago • 2 comments

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

akari0216 avatar May 19 '22 02:05 akari0216

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

cgnerds avatar May 19 '22 15:05 cgnerds

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

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

akari0216 avatar May 24 '22 10:05 akari0216