def _register_base_optimizer(key, fn): registry.register_optimizer(key)( lambda learning_rate, hparams: fn(learning_rate))
def _register_base_optimizer(name, opt): key = misc_utils.camelcase_to_snakecase(name) if key in registry.Registries.optimizers: return registry.register_optimizer(key)( lambda learning_rate, hparams: opt(learning_rate))