TensorRT:ShuffLayer && BatchNorm1d

tech2022-10-03  98

正常的网络结构都是 Conv + BN + Relu, 但自己之前项目中搭建的caffe模型是FC + BN的操作,所以转到pytorch就卡了一下

pytorch中FC后面如果接上BN,该怎么操作呢?

使用nn.BatchNorm1d(),而不是nn.BatchNrom2d(),因为给到BN层的数据维度信息是不一样的

如果是Conv + BN,可以看到BN的输入是NCHW

如果是FC + BN, 可以看到BN 的输入是 NC

所以需要调用的不同的BN接口

最新回复(0)