search

HIT-SCIR / plm-nlp-code

Apache License 2.0
620 stars 195 forks source link

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

Open akari0216 opened 2 years ago

akari0216 commented 2 years ago

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

cgnerds commented 2 years ago

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

akari0216 commented 2 years ago

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

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