在pytorch和keras中保存模型

Question

我已经使用keras训练了模型，并在pytorch的帮助下保存了模型。将来会不会引起任何问题。据我所知，它们之间的唯一区别是 Keras 将其模型的权重保存为两倍，而 PyTorch 将其权重保存为浮点数。

Answer 1

您可以通过执行以下操作将模型转换为两倍

model.double（）

请注意，此后，您将需要输入 DoubleTensor 。