meatball1982 发表于 2023-7-29 10:43:50

豆粑粑 pytorch for out of memory

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


    with torch.no_grad():
      = model(input)


可以解决。
页: [1]
查看完整版本: 豆粑粑 pytorch for out of memory