    def test_conv3d_bn_activation(self):
        """
        Test conv2d bn activation  layer.
        """
        x = torch.randn(1, 3, 3, 4, 4)
        model = Conv3DBNActivation(
            in_planes=3,
            out_planes=3,
            kernel_size=(2, 2, 2),
            padding=(1, 1, 1),
            stride=(1, 1, 1),
            norm_layer=nn.BatchNorm3d,
            activation_layer=nn.ReLU
        )
        y = model(x)
        # print(y.shape)
        self.assertTrue(y.shape == torch.Size([1, 3, 4, 5, 5]))
Test conv2d bn activation  layer.
tests/unittests/models/layers/test_conv_bn_activation.py