From 12e330cd2071c708822bf541ffbd3a56b5c38f9e Mon Sep 17 00:00:00 2001 From: lilei Date: Sat, 27 Jun 2020 17:22:50 +0800 Subject: [PATCH] fix group param not list --- mindspore/nn/optim/optimizer.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mindspore/nn/optim/optimizer.py b/mindspore/nn/optim/optimizer.py index cdda93ab49..a811edcabc 100755 --- a/mindspore/nn/optim/optimizer.py +++ b/mindspore/nn/optim/optimizer.py @@ -262,6 +262,10 @@ class Optimizer(Cell): if not group_param['params']: raise ValueError("Optimizer got an empty group parameter list.") + for param in group_param['params']: + if not isinstance(param, Parameter): + raise TypeError("The group param should be an iterator of Parameter type.") + def _parse_group_params(self, parameters, learning_rate): """Parse group params.""" self._check_group_params(parameters)