豆粑粑 pytorch for out of memory

meatball1982

用pytorch训练，验证模型，
在test阶段，通过迭代实现预测。总是out of memory.
提示显存不够，
2023年7月。A100,32G显示，也经常性的就不够。


    with torch.no_grad():
        [loc_pred,out_prev]= model(input)


可以解决。