鱼C论坛's Archiver
论坛
›
新手乐园
› 模型测试的时候爆显存
Handsome_zhou
发表于 2022-11-20 01:46:23
模型测试的时候爆显存
加载训练好的模型进行测试的时候提示显存不足:
可能是忘记加: with torch.no_grad():
页:
[1]
查看完整版本:
模型测试的时候爆显存