网站首页编程语言正文

Assertion failed: cond.is_weights() && cond.weights().count() == 1 && “If condition must be a initia

作者：xuan_gxi 更新时间： 2022-07-26 编程语言

用onnx–>trt文件时，出现错误:Assertion failed: cond.is_weights() && cond.weights().count() == 1 && "If condition must be a initia
将生成的onnx文件用netron(https://netron.app/)打开，
原因出在这：if语句
在这里插入图片描述
解决办法：实例化网络(pth–onnx)的时候把if语句去掉，找squeeze(1)操作删掉，重新生成onnx即可

原文链接：https://blog.csdn.net/xuan_gap/article/details/125937774

上一篇：ubuntu18.04+cuda10.2+tensorrt8
下一篇：分析查询语句 Explain

网站首页 编程语言 正文

Assertion failed: cond.is_weights() &amp;&amp; cond.weights().count() == 1 &amp;&amp; “If condition must be a initia

相关推荐

网站首页编程语言正文

Assertion failed: cond.is_weights() && cond.weights().count() == 1 && “If condition must be a initia