From ed026519d959ecc60a895f379c228de5df77ffb0 Mon Sep 17 00:00:00 2001
From: zhang <zch921005@126.com>
Date: Sun, 19 Jun 2022 09:20:25 +0800
Subject: daily update

---
 pretrained/swin/demo.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'pretrained/swin/demo.py')

diff --git a/pretrained/swin/demo.py b/pretrained/swin/demo.py
index 6f8d87b..3b6d9d9 100644
--- a/pretrained/swin/demo.py
+++ b/pretrained/swin/demo.py
@@ -20,6 +20,7 @@ if __name__ == '__main__':
     batch_input = transformed.unsqueeze(0)
     print(batch_input.shape)
 
+    timm.list_models()
     model = timm.create_model('swin_base_patch4_window7_224', pretrained=True)
 
     with torch.no_grad():
-- 
cgit v1.2.3