swin_Tranformer使用netron可视化

操作记录和遇到的问题

由于使用netron可视化swin_tiny的时候遇到了一些问题，因此记录一下操作过程和遇到的问题。

准备工作

ubuntu/windows 本人是ubuntu

激活虚拟环境

activate your env :conda activate urenv；

需要用到的package：

1. timm :pip install timm == 0.6.13
2. onnx :conda install onnx
3. netron :pip install netron
4. torch: conda install torch

我的package:

netron=7.6.8=pypi_0
onnx=1.16.0=pypi_0
python=3.10.11=h7a1cb2a_2
pytorch=1.13.1=py3.10_cuda11.6_cudnn8.3.2_0
timm=0.6.13=pypi_0
torch=1.13.1=pypi_0
torchvision=0.14.1=pypi_0

下载模型

从meta官网下载想要的模型
例：swin_tiny的模型

代码

import torch
import timm

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')

modelpath=