pytorchでBachNormalization層を取り除く方法

Question

pytorchでclassの中のnn.BatchNorm2d層を取り除きたいのですが、コメントアウトするとエラーが出ます。

class ConvBNReLU(nn.Sequential):
    def __init__(self, in_planes, out_planes, kernel_size=3, stride=1, dilation=1, groups=1):
        #padding = (kernel_size - 1) // 2
        super(ConvBNReLU, self).__init__(
            nn.Conv2d(in_planes, out_planes, kernel_size, stride, 0, dilation=dilation, groups=groups, bias=False),
            nn.BatchNorm2d(out_planes), # コメントアウトするとエラーになる
            nn.ReLU6(inplace=True)
        )

以下のようにBatchNorm2dを取り除きたいのですが、どのように書いたら良いでしょうか？
やりたいこと

class ConvBNReLU(nn.Sequential):
    def __init__(self, in_planes, out_planes, kernel_size=3, stride=1, dilation=1, groups=1):
        #padding = (kernel_size - 1) // 2
        super(ConvBNReLU, self).__init__(
            nn.Conv2d(in_planes, out_planes, kernel_size, stride, 0, dilation=dilation, groups=groups, bias=False),
            # nn.BatchNorm2d(out_planes),
            nn.ReLU6(inplace=True)
        )

Comments

[tiitoi]

どのようなエラーがでるのでしょうか？
forward() 時にエラーが出るのか、重み読み込みでエラーが出るのか etc...

Answer 1

mobilenetの重みを取り除いたら解決しました