模型转化（.pth--->.pt）

import torch
import torch.utils.data.distributed

# pytorch环境中
model_pth = 'model_31_0.96.pth' #模型的参数文件
mobile_pt ='model.pt' # 将模型保存为Android可以调用的文件

model = torch.load(model_pth)
model.eval() # 模型设为评估模式
device = torch.device('cpu')
model.to(device)
# 1张3通道224*224的图片
input_tensor = torch.rand(1, 3, 224, 224) # 设定输入数据格式

mobile = torch.jit.trace(model, input_tensor) # 模型转化
mobile.save(mobile_pt) # 保存文件

其余：https://blog.csdn.net/y_dd6011/article/details/104751029

https://blog.csdn.net/hhhhhhhhhhwwwwwwwwww/article/details/122860445

标签：pt,pth,mobile,模型,torch,pytorch,安卓端,model,224
来源： https://www.cnblogs.com/hans-learn-python/p/16247606.html