def __init__(self):
    super(bn_3d_model, self).__init__(name='test_bn_3d_model')
    self.conv = Conv3d(n_filter=32, filter_size=(3, 3, 3), strides=(2, 2, 2), name='test_conv3d', in_channels=3)
    self.bn = BatchNorm3d(num_features=32, name='test_bn3d')
def __init__(self):
    super(bn_3d_model, self).__init__(name='test_bn_3d_model')
    self.conv = Conv3d(n_filter=32, filter_size=(3, 3, 3), strides=(2, 2, 2), name='test_conv3d', in_channels=3)
    self.bn = BatchNorm3d(num_features=32, name='test_bn3d')