Browse Source

!2621 fix group params not list

Merge pull request !2621 from lilei/fix
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
f65756162e
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