首页 > 解决方案 > 如何在不访问类代码的情况下加载 pytorch 模型?

问题描述

所以我正在寻找一种方法来加载一些模型而不访问类代码。有办法吗?使这种事情成为可能的唯一方法是使用 torch.jit.save/load 保存模型,但如果我不是训练/设计模型的人怎么办?

使用 tensorflow,我只是习惯于使用 load_model 函数。pytorch 中有类似的东西吗?

标签: pytorch

解决方案


推荐阅读