コード例 #1
0
ファイル: optimize.py プロジェクト: surdomic/tensor2tensor 
def _register_base_optimizer(key, fn):
    registry.register_optimizer(key)(
        lambda learning_rate, hparams: fn(learning_rate))
コード例 #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))