tfm.optimization.register_optimizer_cls

Register customize optimizer cls.

View aliases

Main aliases

tfm.optimization.optimizer_factory.register_optimizer_cls

tfm.optimization.register_optimizer_cls(
    key: str, optimizer_config_cls: tf.keras.optimizers.Optimizer
)

The user will still need to subclass data classes in configs.optimization_config to be used with OptimizerFactory.

Args
key	A string to that the optimizer_config_cls is registered with.
optimizer_config_cls	A class which inherits tf.keras.optimizers.Optimizer.