From 94b78fdf1b750de062b2b1d5f63cd42ab9876260 Mon Sep 17 00:00:00 2001 From: Ziyan Date: Mon, 8 Jun 2020 16:16:15 +0800 Subject: [PATCH] fix_lars_computation_error --- mindspore/nn/optim/lars.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/optim/lars.py b/mindspore/nn/optim/lars.py index 3d85a05867..001a578ffe 100755 --- a/mindspore/nn/optim/lars.py +++ b/mindspore/nn/optim/lars.py @@ -94,7 +94,7 @@ class LARS(Optimizer): self.learning_rate = optimizer.learning_rate self.lars = P.LARSUpdate(epsilon, hyperpara, use_clip) self.reciprocal_scale = 1.0 / loss_scale - self.weight_decay = weight_decay * loss_scale + self.weight_decay = weight_decay self.cast = P.Cast() self.decay_flag = tuple(decay_filter(x) for x in self.parameters) self.lars_flag = tuple(lars_filter(x) for x in self.parameters)