chore: Add efficientnet b0 and VIT to testsuite

peri044 · peri044 · commit e08d1c7b7444 · 2021-06-30T12:46:03.000-07:00
Signed-off-by: Dheeraj Peri &lt;peri.dheeraj@gmail.com&gt;
diff --git a/tests/modules/hub.py b/tests/modules/hub.py
@@ -66,9 +66,13 @@
         "model": models.detection.fasterrcnn_resnet50_fpn(pretrained=True),
         "path": "script"
     },
-    "vit": {
+    "efficientnet_b0": {
         "model": timm.create_model('efficientnet_b0', pretrained=True),
         "path": "script"
+    },
+    "vit": {
+        "model": timm.create_model('vit_base_patch16_224', pretrained=True),
+        "path": "script"
     }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -66,9 +66,13 @@`
`66`	`66`	`"model": models.detection.fasterrcnn_resnet50_fpn(pretrained=True),`
`67`	`67`	`"path": "script"`
`68`	`68`	`},`
`69`		`- "vit": {`
	`69`	`+ "efficientnet_b0": {`
`70`	`70`	`"model": timm.create_model('efficientnet_b0', pretrained=True),`
`71`	`71`	`"path": "script"`
	`72`	`+ },`
	`73`	`+ "vit": {`
	`74`	`+ "model": timm.create_model('vit_base_patch16_224', pretrained=True),`
	`75`	`+ "path": "script"`
`72`	`76`	`}`
`73`	`77`	`}`
`74`	`78`