会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
python我的最爱
坚持每天分享一个深度学习的知识点
博客园
首页
新随笔
联系
订阅
管理
2021年8月18日
算法工具-1.torch Pt模型转onnx(torch.onnx.export(m, d, onnx_path))
摘要: 使用torch.onnx.export来进行模型的构造 import torch import torch.nn as nn import torch.nn.functional as F import torch.onnx import netron class model(nn.Module):
阅读全文
posted @ 2021-08-18 22:48 python我的最爱
阅读(2264)
评论(0)
推荐(0)
编辑