o e5@sdZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Zdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZedddZeddddZ edddZ!dS)zcBuilt-in optimizer classes. For more examples see the base class `tf.keras.optimizers.Optimizer`. )backend) Optimizer) TFOptimizer)adadelta)adagrad)adam)adamax)ftrl)gradient_descent)nadam) optimizer_v2)rmsprop)deserialize_keras_objectserialize_keras_object optimizer) keras_exportzkeras.optimizers.serializecCst|S)a@Serialize the optimizer configuration to JSON compatible python dict. The configuration can be used for persistence and reconstruct the `Optimizer` instance again. >>> tf.keras.optimizers.serialize(tf.keras.optimizers.SGD()) {'class_name': 'SGD', 'config': {'name': 'SGD', 'learning_rate': 0.01, 'decay': 0.0, 'momentum': 0.0, 'nesterov': False}} Args: optimizer: An `Optimizer` instance to serialize. Returns: Python dict which contains the configuration of the input optimizer. rrrRD:\Projects\ConvertPro\env\Lib\site-packages\tensorflow/python/keras/optimizers.py serialize'srzkeras.optimizers.deserializeNc Csjddlm}tjtjtjtj t j t j tjtj|j|jd }|d|vr-|d|d<t|||ddS)a-Inverse of the `serialize` function. Args: config: Optimizer configuration dictionary. custom_objects: Optional dictionary mapping names (strings) to custom objects (classes and functions) to be considered during deserialization. Returns: A Keras Optimizer instance. r)loss_scale_optimizer) rrrrr r Zsgdr ZlossscaleoptimizerZlossscaleoptimizerv1 class_namer)Zmodule_objectscustom_objectsZprintable_module_name)Z'tensorflow.python.keras.mixed_precisionr adadelta_v2ZAdadelta adagrad_v2ZAdagradadam_v2ZAdam adamax_v2ZAdamaxnadam_v2ZNadam rmsprop_v2ZRMSpropgradient_descent_v2ZSGDr ZFtrlZLossScaleOptimizerlowerr)configrrZ all_classesrrr deserialize<s( r#zkeras.optimizers.getcCsvt|ttjfr |St|tjrt|}t||St|tr$t |St|t r4t |id}t |St d |)aRetrieves a Keras Optimizer instance. Args: identifier: Optimizer identifier, one of - String: name of an optimizer - Dictionary: configuration dictionary. - Keras Optimizer instance (it will be returned unchanged). - TensorFlow Optimizer instance (it will be wrapped as a Keras Optimizer). Returns: A Keras Optimizer instance. Raises: ValueError: If `identifier` cannot be interpreted. )rr"z,Could not interpret optimizer identifier: {}) isinstancerr Z OptimizerV2tf_optimizer_modulerrZtrack_tf_optimizerdictr#str ValueErrorformat) identifieroptr"rrrgetes    r,)N)"__doc__Ztensorflow.python.kerasrZ$tensorflow.python.keras.optimizer_v1rrZ$tensorflow.python.keras.optimizer_v2rrrrrrrrr r r r rr r rZ+tensorflow.python.keras.utils.generic_utilsrrZtensorflow.python.trainingrr%Z tensorflow.python.util.tf_exportrrr#r,rrrrs.                  (