TensorRT：ShuffLayer && BatchNorm

時間 2021-01-08

標籤目標檢測-實現简体版

原文原文鏈接

正常的網絡結構都是 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）
更多相關文章...

相關標籤/搜索

batchnorm