TensorFlow 1 version | View source on GitHub |
Provides a scope that changes to _GLOBAL_CUSTOM_OBJECTS
cannot escape.
tf.keras.utils.CustomObjectScope(
*args
)
Code within a with
statement will be able to access custom objects
by name. Changes to global custom objects persist
within the enclosing with
statement. At end of the with
statement,
global custom objects are reverted to state
at beginning of the with
statement.
Example:
Consider a custom object MyObject
(e.g. a class):
with CustomObjectScope({'MyObject':MyObject}):
layer = Dense(..., kernel_regularizer='MyObject')
# save, load, etc. will recognize custom object by name
Methods
__enter__
__enter__()
__exit__
__exit__(
*args, **kwargs
)