我只想在pytorch中使用预先训练的模型,而无需安装整个软件包。
我可以从pytorch复制模型模块吗?
答案 0 :(得分:0)
恐怕您不能这样做:为了运行模型,不仅需要训练的权重('.pth.tar'文件),还需要网络的“结构”:即层,以及它们如何相互连接等。此网络结构使用python编码,需要安装pytorch。
答案 1 :(得分:0)
在不安装PyTorch的情况下使用PyTorch模型的一种方法是将模型以Onnx格式导出。一旦模型为Onnx格式,就可以将模型导入Onnx运行时,并可以用于推理。本教程将为您提供帮助。Pytorch ONNX