o i e 2@sBdZddlZddlZddlmmZddlmZddl m Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddl mZddlm ZddlmZddlmZdd lmZdd lmZddlmZ dd lmZ!ddlm"Z#ddlmZ$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl5m7Z7ddl8m9Z9ddl:m;Z<ddl=m>Z>dd l=m?Z?dd!l@mAZAeAd"d2d$d%ZBd&d'ZCeAd(d3d)d*ZDeAd+gd,d-d.ZEeAd/d0d1ZFdS)4zcBuilt-in optimizer classes. For more examples see the base class `tf.keras.optimizers.Optimizer`. N)logging)backend)adadelta) adafactor)adagrad)adam)adamax)adamw)ftrl)lion)nadam) optimizer)rmsprop)sgd)gradient_descent) optimizer_v2)Adadelta)Adagrad)Adam)Adamax)Ftrl)SGD)Nadam)RMSprop) Optimizer) TFOptimizer)learning_rate_schedule) serialization)deserialize_keras_object)serialize_keras_object) keras_exportzkeras.optimizers.serializeFcCsN |durdSt|tjttjfstdt|d|r#t |St |S)aSerialize 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.legacy.SGD()) {'module': 'keras.optimizers.legacy', 'class_name': 'SGD', 'config': {'name': 'SGD', 'learning_rate': 0.01, 'decay': 0.0, 'momentum': 0.0, 'nesterov': False}, 'registered_name': None}NzThe `keras.optimizers.serialize()` API should only be used for objects of type `keras.optimizers.Optimizer`. Found an instance of type z+, which may lead to improper serialization.) isinstancebase_optimizerrbase_optimizer_legacy OptimizerV2warningswarntypelegacy_serializationr)r use_legacy_formatr*L/var/www/myenv/lib/python3.10/site-packages/keras/src/optimizers/__init__.py serializeJs$  r,cCstdko tdkS)NDarwinarm)platformsystem processorr*r*r*r+ is_arm_macosr2zkeras.optimizers.deserializecKs4ddlm}|dd}|rtd|t|ddkr%|ddd}tjj rVt rVt sV|sVt j tjtjtjt j tjtjtjtjtjtjtj|j|j|jd }n!tj tjt jt!jt j tjtjtjt"jt#jt$jt%j|j|j|jd }|d &|vr|d &|d <|rt'j(|||d d St(|||d d S) aIInverse 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. rloss_scale_optimizeruse_legacy_optimizerFInvalid keyword arguments: configis_legacy_optimizerT)rrrrexperimentaladadeltaexperimentaladagradexperimentaladamexperimentalsgdr rrr lossscaleoptimizerlossscaleoptimizerv3lossscaleoptimizerv1 class_namer )module_objectscustom_objectsprintable_module_name))keras.src.mixed_precisionr4pop TypeErrorlengettf __internal__tf2enabledexecuting_eagerlyr2rrrrrrrrrrr rrrr rLossScaleOptimizerV3LossScaleOptimizeradadelta_legacyadagrad_legacy adam_legacy adamax_legacy nadam_legacyrmsprop_legacygradient_descent_legacy ftrl_legacylowerr(r)r7rBr)kwargsr4r5 all_classesr*r*r+ deserializessz    r[z9keras.__internal__.optimizers.convert_to_legacy_optimizer)v1cCsddlm}t|tjstd|dt|d|jj }| }gd}|D]}| |dq)t||j rEt |j|d<|d krEd }t|d rVt|jtjrV|j|d <||d }t|ddS)a|Convert experimental optimizer to legacy optimizer. This function takes in a `keras.optimizers.Optimizer` instance and converts it to the corresponding `keras.optimizers.legacy.Optimizer` instance. For example, `keras.optimizers.Adam(...)` to `keras.optimizers.legacy.Adam(...)`. Args: optimizer: An instance of `keras.optimizers.Optimizer`. rr3zr`convert_to_legacy_optimizer` should only be called on instances of `tf.keras.optimizers.Optimizer`, but received z of type .) weight_decayuse_ema ema_momentumema_overwrite_frequency jit_compiler8Ninner_optimizerr>r=_learning_rate learning_rater@r7Tr5)rDr4r!r"r ValueErrorr' __class____name__rX get_configrErNconvert_to_legacy_optimizerrchasattrrdrLearningRateScheduler[)r r4optimizer_namer7keys_to_removekeylegacy_optimizer_configr*r*r+rls8       rlzkeras.optimizers.getcKs|dd}|rtd|t|ttjfr|St|tjr+tjj r'|St |St|tj j jjr?t|}t||St|trOd|v}t|||dSt|trat|id}t||dStd|) 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. r5Fr6module)r5r)rfrgz*Could not interpret optimizer identifier: )rErFr!rr#r$r"rIrJrKrLrlcompatr\trainrrtrack_tf_optimizerdictr[strrHrh) identifierrYr5optr)r7r*r*r+rHsD      rH)F)NF)G__doc__r/r%tensorflow.compat.v2rtv2rIabslr keras.srcrkeras.src.optimizersrrrrrr r r r r r"rrkeras.src.optimizers.legacyrPrQrRrSrWrrVrTrr#rU$keras.src.optimizers.legacy.adadeltar#keras.src.optimizers.legacy.adagradr keras.src.optimizers.legacy.adamr"keras.src.optimizers.legacy.adamaxr keras.src.optimizers.legacy.ftrlr,keras.src.optimizers.legacy.gradient_descentr!keras.src.optimizers.legacy.nadamr#keras.src.optimizers.legacy.rmspropr!keras.src.optimizers.optimizer_v1rrkeras.src.optimizers.schedulesrkeras.src.saving.legacyrr("keras.src.saving.serialization_librr tensorflow.python.util.tf_exportr r,r2r[rlrHr*r*r*r+sj                                       $ _ ;