pytorch官方的流程:https://pytorch.org/docs/1.1.0/onnx.html tensorrt中如何新增自定义层 知乎上一篇如何给tensorrt添加自定义层

政安晨的个人主页：政安晨 欢迎 👍点赞✍评论⭐收藏 收录专栏: TensorFlow与Keras实战演绎机器学习 希望政安晨的博客能够对您有所裨益，如有不足之处，欢迎在评论区提出指正！ 介绍 本文将涵盖构建自己的子类化层和模型所需的所有知识。 您将了解以下功能： 层类 add_weight()方法 可训练和不可训练的权重 build()方法 确保您的层可以与任何后端一起使

真的是被Caffe玩哭啦！。先说一下我的情况吧。我是用的Caffe是Windows版本，也许Linux版本就没有我的烦恼了。我想在训练的时候使用 BatchNormail层，由于我原先使用的是大神happynear的老版本了，但是老版本里没有Scale层，所以只能更新新版本。于是我使用官方Caffe-Microsoft版本，训练还算一切正常。但是到了应用阶段出现问题了。在命名空间caffe中找不到