train.py 文件源码

def get_optimizer_from_config(model, config):
    opt_config = Optimizer(**config['optimizer'])
    optimizer = getattr(optimizers, opt_config.method)(**opt_config.args)
    optimizer.setup(model)
    if opt_config.weight_decay is not None:
        optimizer.add_hook(
            chainer.optimizer.WeightDecay(opt_config.weight_decay))
    return optimizer