如何加载子类keras模型

Question

我正在tensorflow 2.4.1和Ubuntu20.04上运行这个tutorial

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

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

当我使用以下命令加载模型时：

tf.saved_model.load("saved_model_dir/")

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