Handsome_zhou 发表于 2022-11-20 01:46:23

模型测试的时候爆显存

加载训练好的模型进行测试的时候提示显存不足:
       
        可能是忘记加: with torch.no_grad():
页: [1]
查看完整版本: 模型测试的时候爆显存