Add ViT to sotabench

7613094f · Ross Wightman · 477a78ed · 7613094f
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

sotabench.py sotabench.py +3 -0

未找到文件。
--- a/sotabench.py
+++ b/sotabench.py
@@ -458,6 +458,9 @@ model_list = [
    _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():