错误如题所述,我也尝试了网上其他人的方法,但是可能人家的项目代码相对简单,比较容易找到output = net(…)以及x_train 中的requires_grad=True。可能是我的代码层层调用的关系,反正我是找了半天也找不到这方面的东西,修改了一些我认为类似的但是最后发现没什么用。
一开始我是将max_sentences由64设置成32,但是就算设置到4,发现仍然会报错。后来在同事的指导下把max_sentences调小后,又把max_sentences_valid设置为了16,然后再跑一下就跑通了。如下图:
关于错误runtimeerror: CUDA out of memory. Tried to allocate 1.17 GB
最新推荐文章于 2024-06-14 16:03:35 发布