After recent upgrade to Tensorflow 2.3 i cannot save TF-agents layers, i get this:
AttributeError: \'ActorDistributionNetwork\' object has no attribute \'save_wei
