TensorRT：ShuffLayer && BatchNorm
正常的网络结构都是 Conv + BN + Relu， 但自己之前项目中搭建的caffe模型是FC + BN的操作，所以转到pytorch就卡了一下 pytorch中FC后面如果接上BN，该怎么操作呢？ 使用nn.BatchNorm1d()，而不是nn.BatchNrom2d()，因为给到BN层的数据维度信息是不一样的 如果是Conv + BN，可以看到BN的输入是NCHW 如果是FC + BN，
相关文章
- 1. Pytorch BatchNorm
- 2. BatchNorm（Pytorch ）
- 3. BatchNorm的理解
- 4. caffe Batchnorm使用
- 5. BatchNorm VS InstanceNorm
- 6. BatchNorm(BN)方法
- 7. batchnorm解读
- 8. 批归一化 BatchNorm
- 9. Pytorch(笔记5) -- BatchNorm
- 10. 批标准化（BatchNorm）
- 更多相关文章...
相关标签/搜索
每日一句
-
每一个你不满意的现在，都有一个你没有努力的曾经。
欢迎关注本站公众号,获取更多信息
相关文章