Toybrick

EfficientNet.pth转rknn失败

殷艳坤

注册会员

积分
63
楼主
发表于 2023-7-19 10:58:09    查看: 1151|回复: 1 | [复制链接]    打印 | 只看该作者
本帖最后由 殷艳坤 于 2023-7-19 11:41 编辑

E build: Catch exception when building RKNN model!
E build: Traceback (most recent call last):
E build:   File "rknn/api/rknn_base.py", line 1546, in rknn.api.rknn_base.RKNNBase.build
E build:   File "rknn/api/graph_optimizer.py", line 1215, in rknn.api.graph_optimizer.GraphOptimizer.fuse_ops
E build: IndexError: list index (0) out of range

截取一部分网络仍然报上面的错误

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

殷艳坤

注册会员

积分
63
沙发
 楼主| 发表于 2023-7-19 11:44:27 | 只看该作者
已解决:
torch.onnx._export(
        model,
        dummy_input,
        "efficient_b0_dim64.onnx",
        input_names=["input"],
        output_names=["output"],
        do_constant_folding = False,
        opset_version=12,
    )
转onnx加上 do_constant_folding = False,再转就可以了,有哪位大神可以解释下原因吗?do_constant_folding表示是否折叠常量
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表