From ed026519d959ecc60a895f379c228de5df77ffb0 Mon Sep 17 00:00:00 2001 From: zhang Date: Sun, 19 Jun 2022 09:20:25 +0800 Subject: daily update --- pretrained/swin/603_timm_swin.ipynb | 3 +-- pretrained/swin/demo.py | 1 + 2 files changed, 2 insertions(+), 2 deletions(-) (limited to 'pretrained') diff --git a/pretrained/swin/603_timm_swin.ipynb b/pretrained/swin/603_timm_swin.ipynb index 5e4ef1c..2d0cbf2 100644 --- a/pretrained/swin/603_timm_swin.ipynb +++ b/pretrained/swin/603_timm_swin.ipynb @@ -320,8 +320,7 @@ "ExecuteTime": { "end_time": "2022-06-03T03:33:39.593106Z", "start_time": "2022-06-03T03:33:39.583112Z" - }, - "collapsed": true + } }, "outputs": [ { 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