TensorFlow 1 version
|
View source on GitHub
|
Provides a scope that changes to _GLOBAL_CUSTOM_OBJECTS cannot escape.
tf.keras.utils.custom_object_scope(
*args
)
Convenience wrapper for CustomObjectScope.
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
with custom_object_scope({'MyObject':MyObject}):
layer = Dense(..., kernel_regularizer='MyObject')
# save, load, etc. will recognize custom object by name
Arguments | |
|---|---|
*args
|
Variable length list of dictionaries of name, class pairs to add to custom objects. |
Returns | |
|---|---|
Object of type CustomObjectScope.
|