Skip to content

Latest commit

 

History

History

optimizers

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
tests
tests
 
 
BUILD
BUILD
 
 
README.md
README.md
 
 
__init__.py
__init__.py
 
 
adabelief.py
adabelief.py
 
 
average_wrapper.py
average_wrapper.py
 
 
cocob.py
cocob.py
 
 
conditional_gradient.py
conditional_gradient.py
 
 
constants.py
constants.py
 
 
cyclical_learning_rate.py
cyclical_learning_rate.py
 
 
discriminative_layer_training.py
discriminative_layer_training.py
 
 
lamb.py
lamb.py
 
 
lazy_adam.py
lazy_adam.py
 
 
lookahead.py
lookahead.py
 
 
moving_average.py
moving_average.py
 
 
novograd.py
novograd.py
 
 
proximal_adagrad.py
proximal_adagrad.py
 
 
rectified_adam.py
rectified_adam.py
 
 
stochastic_weight_averaging.py
stochastic_weight_averaging.py
 
 
utils.py
utils.py
 
 
weight_decay_optimizers.py
weight_decay_optimizers.py
 
 
yogi.py
yogi.py
 
 

README.md

Addons - Optimizers

Components

https://www.tensorflow.org/addons/api_docs/python/tfa/optimizers

Contribution Guidelines

Standard API

In order to conform with the current API standard, all optimizers must:

  • Inherit from either keras.optimizer_v2.OptimizerV2 or its subclasses.
  • Register as a keras global object so it can be serialized properly: @tf.keras.utils.register_keras_serializable(package='Addons')

Testing Requirements

  • To run your tf.functions in eager mode and graph mode in the tests, you can use the @pytest.mark.usefixtures("maybe_run_functions_eagerly") decorator. This will run the tests twice, once normally, and once with tf.config.run_functions_eagerly(True).

Documentation Requirements