Add ViT to sotabench

rwightman · rwightman · commit 7613094fb5cb · 2020-10-22T16:07:22.000-07:00
diff --git a/sotabench.py b/sotabench.py
@@ -458,6 +458,9 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
     _entry('rexnet_130', 'ReXNet-1.3x', '2007.00992'),
     _entry('rexnet_150', 'ReXNet-1.5x', '2007.00992'),
     _entry('rexnet_200', 'ReXNet-2.0x', '2007.00992'),
+
+    _entry('vit_small_patch16_224', 'ViT-S/16', None),
+    _entry('vit_base_patch16_224', 'ViT-B/16', None),
 ]
 
 if is_server():

Original file line number	Diff line number	Diff line change
`@@ -458,6 +458,9 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,`
`458`	`458`	`_entry('rexnet_130', 'ReXNet-1.3x', '2007.00992'),`
`459`	`459`	`_entry('rexnet_150', 'ReXNet-1.5x', '2007.00992'),`
`460`	`460`	`_entry('rexnet_200', 'ReXNet-2.0x', '2007.00992'),`
	`461`	`+`
	`462`	`+ _entry('vit_small_patch16_224', 'ViT-S/16', None),`
	`463`	`+ _entry('vit_base_patch16_224', 'ViT-B/16', None),`
`461`	`464`	`]`
`462`	`465`
`463`	`466`	`if is_server():`