chore: Add efficientNet to testsuite

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

tests/modules/hub.py

@@ -2,6 +2,7 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torchvision.models as models
+import timm
 
 models = {
     "alexnet": {
@@ -64,6 +65,10 @@
     "faster_rcnn": {
         "model": models.detection.fasterrcnn_resnet50_fpn(pretrained=True),
         "path": "script"
+    },
+    "vit": {
+        "model": timm.create_model('efficientnet_b0', pretrained=True),
+        "path": "script"
     }
 }