【问题标题】:How can I use pytorch pre-trained model without installing pytorch?如何在不安装 pytorch 的情况下使用 pytorch 预训练模型?
【发布时间】:2019-07-19 17:13:21
【问题描述】:

我只想在 pytorch 中使用预训练模型而不安装整个包。
我可以从 pytorch 中复制模型模块吗?

【问题讨论】:

标签: pytorch


【解决方案1】:

在不安装 PyTorch 的情况下使用 PyTorch 模型的一种方法是以 Onnx 格式导出模型。一旦模型采用 Onnx 格式,模型就可以导入 Onnx 运行时并用于推理。本教程应该可以帮到你。Pytorch ONNX

【讨论】:

  • 这可能是一个有趣的建议,但现在这个答案显示在审核队列中,因为它非常短,它实际上由一个链接组成。由于它可能最终以当前形式成为deleted,因此请考虑使用一些详细信息和步骤扩展答案。例如,一个有趣的信息是知道 ONNX 的安装是否比 PyTorch 轻得多。
【解决方案2】:

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

【讨论】:

    猜你喜欢
    • 2021-03-08
    • 1970-01-01
    • 2019-09-11
    • 2018-02-20
    • 1970-01-01
    • 2021-07-16
    • 1970-01-01
    • 2021-04-05
    • 2021-07-01
    相关资源
    最近更新 更多