    def test_Movinet_A0(self):
        """
        Test MovinetA0 model.
        """
        model = create_model(model_name='movineta0',causal=False)
        x = torch.randn(1, 3, 1, 64, 64)
        if torch.cuda.is_available():
            x = x.cuda()
            model = model.cuda()
        y = model(x)
        self.assertTrue(y.shape == torch.Size([1, 600]))
Test MovinetA0 model.
tests/unittests/models/movinet/test_movinet.py