From 43c7eec53690fed759460730821eefd96170343a Mon Sep 17 00:00:00 2001 From: Neel Kovelamudi Date: Wed, 18 Jan 2023 14:10:37 -0800 Subject: [PATCH] Changes external references to `keras.utils.serialize_keras_object/deserialize_keras_object` to legacy serialization API in preparation for switching all of Keras to new serialization format. PiperOrigin-RevId: 502975348 --- tensorflow_graphics/projects/gan/keras_layers.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tensorflow_graphics/projects/gan/keras_layers.py b/tensorflow_graphics/projects/gan/keras_layers.py index 4580dbdb1..c74fedd6c 100644 --- a/tensorflow_graphics/projects/gan/keras_layers.py +++ b/tensorflow_graphics/projects/gan/keras_layers.py @@ -393,14 +393,14 @@ def call( def get_config(self) -> Dict[str, Any]: """Returns the config of the layer.""" config = { - 'kernel_size': - self._kernel_size, - 'filters': - self._filters, - 'kernel_initializer': - tf.keras.utils.serialize_keras_object(self._kernel_initializer), - 'bias_initializer': - tf.keras.utils.serialize_keras_object(self._bias_initializer), + 'kernel_size': self._kernel_size, + 'filters': self._filters, + 'kernel_initializer': tf.keras.utils.legacy.serialize_keras_object( + self._kernel_initializer + ), + 'bias_initializer': tf.keras.utils.legacy.serialize_keras_object( + self._bias_initializer + ), } base_config = super().get_config() base_config.update(config)