    def forward(self, x: torch.Tensor, input_type: Optional[str] = None):
        """
        Args:
            x: input to the model of shape 1 x C x T x H x W
            input_type: Optional[str] one of ["image", "video", "rgbd"]
                if self.multimodal_model iss True
        Returns:
            preds: tensor of shape (1, num_classes)
        """
        features = self.forward_features(x)
        return self.head(features, input_type = input_type)
Args:
x: input to the model of shape 1 x C x T x H x W
input_type: Optional[str] one of ["image", "video", "rgbd"]
if self.multimodal_model iss True
Returns:
preds: tensor of shape (1, num_classes)
towhee/models/omnivore/omnivore.py