如何在不安装pytorch的情况下使用pytorch预训练模型?

时间:2019-02-26 07:59:10

标签: pytorch

我只想在pytorch中使用预先训练的模型,而无需安装整个软件包。
我可以从pytorch复制模型模块吗?

2 个答案:

答案 0 :(得分:0)

恐怕您不能这样做:为了运行模型,不仅需要训练的权重('.pth.tar'文件),还需要网络的“结构”:即层,以及它们如何相互连接等。此网络结构使用python编码,需要安装pytorch。

答案 1 :(得分:0)

在不安装PyTorch的情况下使用PyTorch模型的一种方法是将模型以Onnx格式导出。一旦模型为Onnx格式,就可以将模型导入Onnx运行时,并可以用于推理。本教程将为您提供帮助。Pytorch ONNX