最近用Pytorch跑模型,在自己电脑上跑的挺好的,但是一送到服务器上去跑就挂了。一开始以为是batchsize开太大了爆炸了,后面调小了batchsize依旧爆炸,才意识到可能不是这里的问题,应该是出现内存泄漏了。 于是我边跑模型边盯着服务器的RA…
此系列所有笔记来源参考: [github author="lyhue1991" project="eat_pytorch_in_20_days"][/github] 上节中对Titanic数据集利用pytorch做了简单的预测,本节中我们开始从pyt…