def get_optimizer(model, method, optimizer_args, weight_decay=None):
optimizer = getattr(optimizers, method)(**optimizer_args)
optimizer.setup(model)
if weight_decay is not None:
optimizer.add_hook(chainer.optimizer.WeightDecay(weight_decay))
return optimizer
评论列表
文章目录