【发布时间】:2018-10-13 19:38:24
【问题描述】:
有没有办法在训练后保存模型并与其他人共享模型?像普通脚本一样?由于网络是浮点矩阵的集合,是否可以仅提取这些训练好的权重并在新数据上运行以进行预测,而不是要求用户也安装这些框架?我是这些框架的新手,我会根据需要做出任何澄清。
【问题讨论】:
标签: tensorflow keras pytorch
有没有办法在训练后保存模型并与其他人共享模型?像普通脚本一样?由于网络是浮点矩阵的集合,是否可以仅提取这些训练好的权重并在新数据上运行以进行预测,而不是要求用户也安装这些框架?我是这些框架的新手,我会根据需要做出任何澄清。
【问题讨论】:
标签: tensorflow keras pytorch
PyTorch:如this post 中所述,您可以将模型的参数保存为字典,或加载字典来设置模型的参数。 您还可以将 PyTorch 模型保存/加载为对象。 这两个程序都要求用户安装至少一个张量计算框架,例如用于高效的矩阵乘法。
【讨论】: