python - 如何加载子类keras模型

Question

我 在 tensorflow 2.4.1 和 ubuntu 20.04 上运行本教程

我使用以下方法保存了模型：

tf.saved_model.save(model, "saved_model_dir/")

当我使用以下方法加载模型时：

tf.saved_model.load("saved_model_dir/")

方法encode()、、decode()和不可用sample()！！！reparameterize()

标签： pythontensorflowkeras