【发布时间】:2019-07-19 17:13:21
【问题描述】:
我只想在 pytorch 中使用预训练模型而不安装整个包。
我可以从 pytorch 中复制模型模块吗?
【问题讨论】:
-
@ghost2718 建议检查 ONNX,pytorch.org/tutorials/advanced/… - 因为答案本身可能会因为仅链接而被删除,所以链接保存在这里。
标签: pytorch
我只想在 pytorch 中使用预训练模型而不安装整个包。
我可以从 pytorch 中复制模型模块吗?
【问题讨论】:
标签: pytorch
在不安装 PyTorch 的情况下使用 PyTorch 模型的一种方法是以 Onnx 格式导出模型。一旦模型采用 Onnx 格式,模型就可以导入 Onnx 运行时并用于推理。本教程应该可以帮到你。Pytorch ONNX
【讨论】:
恐怕你不能这样做:为了运行模型,你不仅需要训练好的权重('.pth.tar' 文件),还需要网络的“结构”:即层,它们如何相互连接等等。这个网络结构是用python编码的,需要安装pytorch。
【讨论】: