日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學(xué)無先后，達(dá)者為師

網(wǎng)站首頁 編程語言 正文

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

作者：xuan_gxi 更新時(shí)間： 2022-07-26 編程語言

用onnx–>trt文件時(shí)，出現(xiàn)錯(cuò)誤:Assertion failed: cond.is_weights() && cond.weights().count() == 1 && "If condition must be a initia
將生成的onnx文件用netron(https://netron.app/)打開，
原因出在這：if語句
在這里插入圖片描述
解決辦法：實(shí)例化網(wǎng)絡(luò)(pth–onnx)的時(shí)候把if語句去掉，找squeeze(1)操作刪掉，重新生成onnx即可

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

欄目分類
最近更新