pytorch吧 关注:1,456贴子:2,583
  • 1回复贴,共1

pytorch 转onnx

只看楼主收藏回复



大佬们,我的模型训练出来可以用,但是转换成onnx后就不行了,在使用onnx推理时会有警告CleanUnusedInitializers] Removing initializer 'layer3.1.bn1.num_batches_tracked'. It is not used by any node and should be removed from the model,这种情况是我后处理错了,还是我转Onnx错了啊?


IP属地:湖北1楼2021-04-15 15:34回复
    应该都没有错 可能是你转的时候有点问题


    2楼2021-05-07 18:16
    回复