pytorch out of memory while validation
使用pytorch0.4框架,在train的时候不会报错,在validation时候报错out of memory错误。
解决方法:
在train过程前需要设置如下:
torch.set_grad_enabled(True)
# switch to train mode
model.train()
在validation过程前需要设置如下:
# In PyTorch 0.4, "v...
原创
2018-12-06 10:23:49 ·
537 阅读 ·
0 评论