Microsoft C++ 异常: c10::Error解决方案

torch::jit::IValue output = module.forward(inputs);  

未经处理的异常
0x00007FFD1F063B19 处(位于 YOLOv5LibTorch.exe 中)有未经处理的异常: Microsoft C++ 异常: c10::Error,位于内存位置 0x000000C6DF4F8EF0 处。

Microsoft C++ 异常: c10::Error解决方案_第1张图片
解决方案引用这篇文章的评论区:https://blog.csdn.net/zzz_zzz12138/article/details/109138805
Microsoft C++ 异常: c10::Error解决方案_第2张图片
如其所说,我也是模型能跑出第一张,在第二张图片就发生错误。
采用将模型clone,

torch::jit::script::Module module2 = module.clone();

在跑完一张图片后再把模型改回来。

module = module2.clone();

即解决问题。
初步判断,应该是跑完一张图片之后,模型发生改变,导致下一张图出错。

你可能感兴趣的:(c++,c++,visual,studio,pytorch)