2024年12月30日
pytorch构建并保存模型(.pth) 转化为 torchscript(.pt), 导出为onnx格式
摘要： pytorch(.pth)模型转化为 torchscript(.pt), 导出为onnx格式 1 .pth模型转换为.pt模型 import torch import torchvision from models import fcn model=torchvision.models.vgg16( 阅读全文
posted @ 2024-12-30 18:31 michaelchengjl 阅读(1027) 评论(0) 推荐(0)