def __init__(self, model=None, lr=0.045, momentum=0.9, weight_decay=4.0e-5):
super(OptimizerGooglenetV2, self).__init__(model)
optimizer = optimizers.MomentumSGD(lr, momentum)
weight_decay = chainer.optimizer.WeightDecay(weight_decay)
optimizer.setup(self.model)
optimizer.add_hook(weight_decay)
self.optimizer = optimizer
self.lr = lr
self.momentum = momentum
self.weight_decay = weight_decay
评论列表
文章目录