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）
- 更多相關文章...
相關標籤/搜索
每日一句
-
每一个你不满意的现在，都有一个你没有努力的曾经。
歡迎關注本站公眾號,獲取更多信息
相關文章