Toybrick

标题: EfficientNet.pth转rknn失败 [打印本页]

作者: 殷艳坤 时间: 2023-7-19 10:58
标题: EfficientNet.pth转rknn失败
本帖最后由殷艳坤于 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
[attach]2730[/attach]
截取一部分网络仍然报上面的错误

作者: 殷艳坤 时间: 2023-7-19 11:44
已解决：
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表示是否折叠常量

欢迎光临 Toybrick (https://t.rock-chips.com/)