TF keras图层不再可保存？

如何解决TF keras图层不再可保存？

最近升级到Tensorflow 2.3后，我无法保存TF-agents图层，我得到了：

AttributeError: 'ActordistributionNetwork' object has no attribute 'save_weights'

由于ActordistributionNetwork是tf.keras.layers.Layer的子类，因此是否删除了单独的keras图层保存自身的功能？在tensorflow和tf-agents的发行变更中，我都找不到关于此的任何信息。 对于model.save_weights，使用tf-agents不太方便，因为我必须为自定义代理使用不同的图层组合。