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

學無先后,達者為師

網站首頁 編程語言 正文

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

欄目分類
最近更新