class_docstr: Test TimeSformer model
code: def test_timesformer_96(self):
      video = torch.randn(1, 3, 4, 4, 4)
      model = timesformer.create_model(
          model_name="timesformer_k400_96x224",
          pretrained=False,
          img_size=4,
          patch_size=2,
      )
      pred = model(video)
      feats = model.forward_features(video)
      self.assertTrue(pred.shape == (1, 400))
      self.assertTrue(feats.shape == (1, 768))
  
file_path: towhee/tests/unittests/models/timesformer/test_timesformer.py
imports: import unittest
  import torch
  from towhee.models import timesformer
label_desc: Use image node_id 2 for calling the TestTimesformer obj's underlying member method code with example usage: obj.test_timesformer_96() without return types