diff options
| author | zhang <zch921005@126.com> | 2022-06-19 09:20:25 +0800 |
|---|---|---|
| committer | zhang <zch921005@126.com> | 2022-06-19 09:20:25 +0800 |
| commit | ed026519d959ecc60a895f379c228de5df77ffb0 (patch) | |
| tree | c2fc265144824c4824814c003c6e0403ebaa2d8c /pretrained/swin/demo.py | |
| parent | 2a42272daeab92ab26481745776dc51ed144924f (diff) | |
daily update
Diffstat (limited to 'pretrained/swin/demo.py')
| -rw-r--r-- | pretrained/swin/demo.py | 1 |
1 files changed, 1 insertions, 0 deletions
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(): |