我真的太难了啊 2022-02-09 16:54 采纳率: 62.5%
浏览 8
已结题

pytorch转caffe

segnet的模型pytorch转caffe时，在 max_pool2d 层遇到了报错 ，查了资料因为，segnet里的 max_pool2d 的参数 return_indices 为 True，目前caffe中不支持，请问大家有没有解决过这个问题的、

  • 写回答

1条回答 默认 最新

  • 爱晚乏客游 2022-02-10 09:32
    关注

    忘记在哪里看到了，好像是要自定义这个最大池化层，不能用torch带的，就可以转出来了

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题？

问题事件

  • 系统已结题 6月7日
  • 已采纳回答 5月30日
  • 创建了问题 2月9日