HRNet的model_best.pth.tar在线转onnx
收藏回复举报
HRNet的model_best.pth.tar在线转onnx
t('forum.solved') 已解决
新人帖
发表于2023-09-15 17:24:32
0 查看

根据https://gitee.com/ascend/ModelZoo-PyTorch/tree/master/PyTorch/contrib/cv/classification/HRNet_ID1780_for_PyTorch官方文档指导,在转onnx的时候,直接跳过加载流程,无法获取到对应的onnx文件。cke_1766.png

加载checkpoint后,无法进行export。注释后科正常导出,但是模型精度不对。

我要发帖子