【问题标题】:How to save and use a trained neural network developed in PyTorch / TensorFlow / Keras?如何保存和使用在 PyTorch / TensorFlow / Keras 中开发的经过训练的神经网络?
【发布时间】:2018-10-13 19:38:24
【问题描述】:

有没有办法在训练后保存模型并与其他人共享模型?像普通脚本一样?由于网络是浮点矩阵的集合,是否可以仅提取这些训练好的权重并在新数据上运行以进行预测,而不是要求用户也安装这些框架?我是这些框架的新手,我会根据需要做出任何澄清。

【问题讨论】:

    标签: tensorflow keras pytorch


    【解决方案1】:

    PyTorch:如this post 中所述,您可以将模型的参数保存为字典,或加载字典来设置模型的参数。 您还可以将 PyTorch 模型保存/加载为对象。 这两个程序都要求用户安装至少一个张量计算框架,例如用于高效的矩阵乘法。

    【讨论】:

      猜你喜欢
      • 2016-12-16
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2017-05-23
      • 2021-07-28
      • 1970-01-01
      • 1970-01-01
      • 2016-12-23
      相关资源
      最近更新 更多