class_docstr: Test MPViT model
class_docstr_tok: ["Test","MPViT","model"]
code: def test_forward(self):
      img = torch.rand(1, 3, 224, 224).to(self.device)
      out = self.model(img)
      print(out.shape)
      self.assertTrue(out.shape == (1, self.num_classes))
  
imports: import unittest
  import torch
  from towhee.models.mpvit import create_model
label_desc: Use image node_id 1 for calling the TestMPViT obj's underlying member method code with example usage: obj.test_forward() without return types