EfficientNet.pth转rknn失败

[复制链接] · 发表于 2023-7-19 10:58:09

本帖最后由殷艳坤于 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

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

只看该作者 · 发表于 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表示是否折叠常量

EfficientNet.pth转rknn失败

本帖子中包含更多资源