Pytorch内存溢出,Ubuntu进程killed
pytorch显存越来越多的一个原因
optimizer.zero_grad()
loss.backward()
optimizer.step()
train_loss += loss
参考了别人的代码发现那句loss一般是这样写
loss_sum += loss.data[0]
这是因为输出的loss的数据类型是Variable。而PyTorch的动态图机制就是通过Variable来构建图。...
原创
2019-12-15 15:48:41 ·
3567 阅读 ·
0 评论