【问题标题】:Saving model in pytorch and keras在 pytorch 和 keras 中保存模型
【发布时间】:2020-11-15 02:33:28
【问题描述】:

我用keras 训练了模型,并在pytorch 的帮助下保存了模型。以后会不会有什么问题。据我所知,它们之间的唯一区别是 Keras 将其模型的权重保存为双精度数,而 PyTorch 将其权重保存为浮点数。

【问题讨论】:

  • 请提供您用于将keras模型保存为pytorch格式的代码。除此之外,在保存模型权重时,它们都有自己的特定格式,并且它们不可互换。但是有一些方法可以将它们相互转换。
  • 1- 请提供一些您实际在做什么的代码。 2-您是否尝试过重新加载模型(最好在其他人的计算机中)?

标签: python machine-learning keras pytorch


【解决方案1】:

您可以通过以下操作将模型转换为双精度

model.double()

请注意,在此之后,您需要输入 DoubleTensor

【讨论】:

    猜你喜欢
    • 2021-12-15
    • 2019-09-26
    • 2021-07-15
    • 1970-01-01
    • 2018-06-25
    • 2020-04-19
    • 2019-04-12
    • 2021-06-13
    • 2021-06-23
    相关资源
    最近更新 更多