将模型保存在gpflow 2中

时间:2020-05-12 12:42:46

标签: tensorflow machine-learning tensorflow2.0 gpflow

我正在尝试保存GPflow模型(在GPflow 2.0版中)。

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

由于gpflow软件包不再包含saver模块,因此有人可以通过其他方法帮助我吗?

1 个答案:

答案 0 :(得分:3)

保存GPflow模型的方式有很多,具体方法取决于您的用例。您可以使用TensorFlow的检查点(保存经过训练的权重),也可以使用TensorFlow的SavedModel格式(保存权重和计算图的一部分)。您可以在intro to GPflow2 notebook中看到这两种方法的示例。