File tree 1 file changed +8
-8
lines changed
tensorflow_graphics/projects/gan
1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -393,14 +393,14 @@ def call(
393
393
def get_config (self ) -> Dict [str , Any ]:
394
394
"""Returns the config of the layer."""
395
395
config = {
396
- 'kernel_size' :
397
- self ._kernel_size ,
398
- 'filters' :
399
- self ._filters ,
400
- 'kernel_initializer' :
401
- tf .keras .utils .serialize_keras_object (self . _kernel_initializer ),
402
- 'bias_initializer' :
403
- tf . keras . utils . serialize_keras_object ( self . _bias_initializer ),
396
+ 'kernel_size' : self . _kernel_size ,
397
+ 'filters' : self ._filters ,
398
+ 'kernel_initializer' : tf . keras . utils . legacy . serialize_keras_object (
399
+ self ._kernel_initializer
400
+ ),
401
+ 'bias_initializer' : tf .keras .utils .legacy . serialize_keras_object (
402
+ self . _bias_initializer
403
+ ),
404
404
}
405
405
base_config = super ().get_config ()
406
406
base_config .update (config )
You can’t perform that action at this time.
0 commit comments