![]() |
Register customize 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. |