您好,欢迎访问代理记账网站
移动应用 微信公众号 联系我们

咨询热线 -

电话 15988168888

联系客服
  • 价格透明
  • 信息保密
  • 进度掌控
  • 售后无忧

Pytorch 模型转化为torchscript 模型——用于C++部署

import torch
import torchvision
import torch.nn as nn
#model = torchvision.models.resnet50(pretrained=True)
model=torchvision.models.resnet18() # 加载模型
num_ftrs = model.fc.in_features
model.fc = nn.Linear(num_ftrs, 196)  # make the change
model.load_state_dict(torch.load("resnet18.pth"))

model.eval()  # 注意,将模型设置为eval模式,再保存

example = torch.rand(1, 3, 224, 224) # 输入模型的尺寸
traced_script_module = torch.jit.trace(model, example)
# 保存模型
traced_script_module.save("resnet18.pt")


分享:

低价透明

统一报价,无隐形消费

金牌服务

一对一专属顾问7*24小时金牌服务

信息保密

个人信息安全有保障

售后无忧

服务出问题客服经理全程跟进