Esempio n. 1
0
def _register_base_optimizer(key, fn):
    registry.register_optimizer(key)(
        lambda learning_rate, hparams: fn(learning_rate))
Esempio n. 2
0
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))