模型训练（6） ResNet18模型可视化

伴生_0904

已于 2024-08-16 19:45:33 修改

阅读量141

点赞数 1

分类专栏：模型训练文章标签：深度学习 pytorch 机器学习

于 2024-08-16 01:28:18 首次发布

本文链接：https://blog.csdn.net/weixin_45705773/article/details/141234856

版权

模型训练专栏收录该内容

6 篇文章 0 订阅

订阅专栏

把resnet18-f37072fd.pth导出为onnx格式（需安装onnx）

import torch
import torchvision.models as models

# 加载您的模型
model = models.resnet18()  # 适当加载和设置您的模型
model.load_state_dict(torch.load('/home/szj/.cache/torch/hub/checkpoints/resnet18-f37072fd.pth'))
model.eval()

# 创建一个示例输入
x = torch.randn(1, 3, 224, 224)  # 自定义输入尺寸

# 导出为 ONNX 格式
torch.onnx.export(model, x, 'resnet18.onnx', verbose=True)

使用netron打开

netron resnet18.onnx