标题: 豆粑粑 pytorch for out of memory [打印本页] 作者: meatball1982 时间: 2023-7-29 10:43 标题: 豆粑粑 pytorch for out of memory 用pytorch训练，验证模型，
在test阶段，通过迭代实现预测。总是out of memory.
提示显存不够，
2023年7月。A100,32G显示，也经常性的就不够。
with torch.no_grad():
[loc_pred,out_prev]= model(input)