神经网络可视化工具Zetane的使用方法

安装Zetane

模型文件格式转换——.pth文件转换为ONNX文件

import torch
import torch.nn as nn
import torchvision
import torchvision.transforms as transforms
from torchvision.models import resnet18

# 设置模型
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = resnet18(pretrained=False, num_classes=10)
model