tensorflow - 将模型保存在 gpflow 2 中

Question

我正在尝试保存 GPflow 模型（在 GPflow 2.0 版中）。

model = gpflow.models.VGP((X, Y_data), kernel=kernel, likelihood=likelihood, num_latent_gps=1)

由于gpflow包不再有saver模块，任何人都可以用另一种方式帮助我吗？

Answer 1

保存 GPflow 模型的方法有多种，具体方法取决于您的用例。您可以使用 TensorFlow 的检查点（保存训练的权重）或使用 TensorFlow 的 SavedModel 格式（保存权重和部分计算图）。您可以在GPflow2 笔记本介绍中看到这两种方法的示例。