torch.onnx.export 時添加參數 training=2,可以將conv和bn 分開顯示,否則onnx默認將bn層融合到conv層。
添加training=2 重新生成onnx之后,再次用Netron 查看網絡結構:
帶有BN層的卷積操作,其卷積的bias 取消掉了,與BN 的beta合並。(https://zhuanlan.zhihu.com/p/353697121)
torch.onnx.export 時添加參數 training=2,可以將conv和bn 分開顯示,否則onnx默認將bn層融合到conv層。
添加training=2 重新生成onnx之后,再次用Netron 查看網絡結構:
帶有BN層的卷積操作,其卷積的bias 取消掉了,與BN 的beta合並。(https://zhuanlan.zhihu.com/p/353697121)
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。