From 4a57948780bb42e59aa2af18c844b1abe09f61b2 Mon Sep 17 00:00:00 2001 From: Neel Kovelamudi Date: Tue, 7 Feb 2023 13:08:31 -0800 Subject: [PATCH] Creates non-breaking changes where necessary in preparation for switching all of Keras to new serialization format. PiperOrigin-RevId: 507864605 --- 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)