pytorch训练/测试注意事项
内存不断增长导致killed
首先检查代码逻辑有无问题,比如有没有出现不断增长的复杂数据结构,比如list,array,dict等等。
torch新老代码不兼容
比如在老版本的torch中,将tensor从gpu转到cpu是这样的
tensor.data.cpu().numpy()
但是在新版本中是
tensor.cpu().detach().numpy()
老版本中让tensor不受梯度的影响
tensor = Variable(tensor.cuda(), volatile=Tr
原创
2021-07-26 19:13:42 ·
215 阅读 ·
0 评论