Browse Source

fix group param not list

tags/v0.6.0-beta
lilei 5 years ago
parent
commit
12e330cd20
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/nn/optim/optimizer.py

+ 4
- 0
mindspore/nn/optim/optimizer.py View File

@@ -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)


Loading…
Cancel
Save