TensorRT – 转换onnx模型出现Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误
1 Error Code 2: Internal Error (Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误
1.1 问题
今天在用TensorRT转化onnx模型出现了以下错误
转换onnx模型出现Error Code 2: Internal Error (Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误
这在https://forums.developer.nvidia.com/t/requires-bool-i-o-but-node-can-not-be-handled-by-myelin/212076 的官网论坛上也提及到了这个问题,主要原因是TensorRT的Slice层不支持bool型的输入
1.2 解决方法
下面两个地方也提到了相同的解决方案:
- https://forums.developer.nvidia.com/t/requires-bool-i-o-but-node-can-not-be-handled-by-myelin/212076
解决方案就是对onnx模型进行操作,在出问题的Slice
前后都插入cast节点进行数值类型转换,将输入到Slice
节点的数据从bool型转换为int32类型,执行完Slice
操作之后再接一个cast节点,从int32类型转换为bool类型。
至于添加cast节点,既可以使用onnx库也可以使用NVIDIA自己的Onnx GraphSurgeon库进行操作。
1.3 使用Onnx GraphSurgeon修改onnx模型,解决转换Slice的输入类型不能为Bool的转换错误
下面是可供参考的使用Onnx GraphSurgeon修改onnx模型,增加Cast节点的代码
(1)请购买正版资源,支持原作者和出版社;
(2)资源收集自互联网,仅供自我学习,请在下载后24小时内删除该资源,如下载者将此资源用于其他非法用途,本站不承担任何法律责任;如有侵权,请立即联系我,马上删除!
(3)此捐赠下载方式无需注册+登录账号,当然如果您喜欢本站也可注册账号;
(4)所支付的款项为捐赠本站款项,而并非付费下载所付款项,本站会在您捐赠本站后显示隐藏内容,请知悉,先非常感谢您对本站的捐赠;
(5)本站支付使用Payjs,该第三方支付方式为微信官方授权的服务商,无支付风险,详情请查看:https://payjs.cn;
(6)请尽量不要使用手机浏览器以及电脑浏览器的无痕模式进行支付操作,以免造成支付成功但未显示隐藏内容。
(7)点击立即购买按钮后弹出选择支付方式页面,选择任意支付方式点击后会跳转到新的二维码支付页面,支付成功后会弹出支付成功的提示框,点击提示框会自动刷新页面显示隐藏内容;另外请不要重复支付,避免财产损失;
(8)在2021年7月27日前,本站采用注册+登录+评论的免费下载模式,旨在节省大家寻找宝贵资源的时间,为大家带来珍贵的知识来源,但是从2021年7月27日起,因遭受注册轰炸和恶意爬虫造成网站负载过大导致宕机,被迫无奈,为减少网站运营时间成本,对部分资源进行捐赠下载,添加支付是反爬虫的最好方式,在此对未享受到免费下载福利的访客表示歉意;
(9)如遇支付页面未弹出、支付成功但未显示隐藏内容、资源链接失效或者提取码出错等情况请发送邮件到stubbornhuang@qq.com或者直接联系QQ:623256052,会尽快为您解决!加我时请备注原因,谢谢!或者点击按钮点击反馈进入反馈页面进行问题反馈
(10)由以上捐赠下载带来的不便请您谅解,如出现重复付费的情况,请联系我,我会尽快返回您的退款!
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:TensorRT – 转换onnx模型出现Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误
原文链接:https://www.stubbornhuang.com/2432/
发布于:2022年12月02日 16:36:35
修改于:2023年06月21日 17:47:33
当前分类随机文章推荐
- TensorRT - TensorRT was linked against cuBLAS/cuBLAS LT 11.6.1 but loaded cuBLAS/cuBLAS LT 11.5.4错误提示 阅读396次,点赞0次
- TensorRT - Polygraphy工具的使用 阅读5757次,点赞0次
- TensorRT - workspace的作用 阅读1349次,点赞0次
- TensorRT - 使用Polygraphy工具比较onnx模型和TensorRT模型的推理结果是否一致 阅读1253次,点赞1次
- TensorRT - 计算模型推理时间 阅读554次,点赞1次
- TensorRT - Invalid Node - TopK,This version of TensorRT only supports input K as an initializer 阅读862次,点赞0次
- TensorRT - 自带工具trtexec的参数使用说明 阅读5501次,点赞0次
- TensorRT - onnx_graphsurgeon工具库的安装与API简介 阅读1957次,点赞0次
- TensorRT - 扩展TensorRT C++API的模型输入维度,增加Dims5,Dims6,Dims7,Dims8 阅读2155次,点赞0次
- TensortRT - 转换模型出现Could not locate zlibwapi.dll. Please make sure it is in your library path!错误 阅读1203次,点赞0次
全站随机文章推荐
- C++ - std::unordered_map中使用结构体或者vector等复杂的数据结构作为Key值 阅读1321次,点赞0次
- 书籍翻译 – Fundamentals of Computer Graphics, Fourth Edition,第7章 Viewing中文翻译 阅读3602次,点赞7次
- 深度学习 - 动作识别Action Recognition最重要的问题 阅读688次,点赞1次
- 我的Windows装机必备软件备忘 阅读573次,点赞0次
- C++11 - 解析并获取可变参数模板中的所有参数 阅读1627次,点赞0次
- 资源分享 - Handbook of Discrete and Computational Geometry, Third Edition英文高清PDF下载 阅读3578次,点赞0次
- TensorRT - 解决INVALID_ARGUMENT: getPluginCreator could not find plugin ScatterND version 1,TensorRT找不到ScatterND插件的问题 阅读5195次,点赞0次
- 书籍翻译 - Fundamentals of Computer Graphics, Fourth Edition 虎书第四版中文翻译 阅读6029次,点赞17次
- 深度学习 - 图解Transformer,小白也能看懂的Transformer处理过程 阅读1072次,点赞0次
- WordPress - 在erphpdown插件中增加在隐藏文章指定位置之后文章内容的付费查看功能 阅读1403次,点赞0次
评论
169