def setup_optimizer(self):
optimizer = optimizers.RMSpropGraves(
lr=self.args.start_lr, alpha=0.95, momentum=0.9, eps=1e-08)
optimizer.setup(self)
optimizer.add_hook(chainer.optimizer.GradientClipping(self.args.grad_clip))
return optimizer
评论列表
文章目录