转换在CUDNNLSTM中训练的模型，以便训练的模型也可以在CPU上运行

Question

我有一个使用CUDNNLSTM训练的模型，我该如何在CPU中使用它？我们如何将CUDNNLSTM变量导出到CPU基本权重，以便训练模型也可以在CPU上运行。

Answer 1

总是有效的一种安全方法是将变量保存到某个文件中，使用适当的格式（如hdf5）并再次手动加载它们。这样您就可以完全控制正在发生的事情，而不依赖于您正在使用的系统。您甚至可以在TensorFlow中训练模型，保存变量并加载它们以便与PyTorch等不同的库一起使用（假设您在那里定义了完全相同的模型）。