plm-nlp-code
plm-nlp-code copied to clipboard
第五章rnnlm.py显存溢出
如标题所示,调整了batch_size=32之后在3060的机器上运行本代码仍然会导致显存溢出,请问有无优化的方法呢?谢谢
可以改成16,我修改后可以运行了。
可以改成16,我修改后可以运行了。
感谢,改成16后确实能跑通了。