From de43c1eefae4ddf0c5b9951316c794d822356979 Mon Sep 17 00:00:00 2001
From: guohongzilong <2713219276@qq.com>
Date: Thu, 2 Jul 2020 17:36:12 +0800
Subject: [PATCH] fix group params order

---
 mindspore/nn/optim/optimizer.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mindspore/nn/optim/optimizer.py b/mindspore/nn/optim/optimizer.py
index 16f252adf..cdf1565f3 100755
--- a/mindspore/nn/optim/optimizer.py
+++ b/mindspore/nn/optim/optimizer.py
@@ -360,16 +360,18 @@ class Optimizer(Cell):
         if len(ordered_parameters) != len(self.group_params):
             raise ValueError(f"The value of 'order_params' should be same with all group parameters.")
 
+        ordered_params = [None] * params_length
         ordered_learning_rate = [None] * params_length
         ordered_weight_decay = [None] * params_length
         params_name = [param.name for param in ordered_parameters]
 
         for param, lr, wd in zip(self.group_params, self.group_lr, self.group_weight_decay):
             index = params_name.index(param.name)
+            ordered_params[index] = param
             ordered_learning_rate[index] = lr
             ordered_weight_decay[index] = wd
 
-        self.group_params = list(ordered_parameters)
+        self.group_params = ordered_params
         self.group_lr = ordered_learning_rate
         self.group_weight_decay = ordered_weight_decay
 
-- 
GitLab