def __init__(self):
    super(bn_2d_model, self).__init__(name='test_bn_2d_model')
    self.conv = Conv2d(n_filter=32, filter_size=(3, 3), strides=(2, 2), name='test_conv2d', in_channels=3)
    self.bn = BatchNorm2d(num_features=32, name='test_bn2d')
def __init__(self):
    super(bn_2d_model, self).__init__(name='test_bn_2d_model')
    self.conv = Conv2d(n_filter=32, filter_size=(3, 3), strides=(2, 2), name='test_conv2d', in_channels=3)
    self.bn = BatchNorm2d(num_features=32, name='test_bn2d')