From 7613094fb5cb960813f606a5c42e3c00c961bc8f Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Thu, 22 Oct 2020 16:07:22 -0700
Subject: [PATCH] Add ViT to sotabench

---
 sotabench.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/sotabench.py b/sotabench.py
index bbc76ba7..d30a7f54 100644
--- 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():