!6199 modified cell.py,solved the problem in yolov3_darknet53(batch_size=16)

Merge pull request !6199 from 吴书全/wsq0914
5 years ago · 07e7f84d46
parent 952135362f 4f0c7a8c45
commit 07e7f84d46
1 changed files with 2 additions and 2 deletions
--- a/mindspore/nn/cell.py
+++ b/mindspore/nn/cell.py
@ -786,8 +786,8 @@ class Cell(Cell_):
            for par_name, par in params:
                if par.inited_param is not None:
                    par = par.inited_param
-                if par is not None and par not in params_set:
-                    params_set.add(par)
+                if par is not None and id(par) not in params_set:
+                    params_set.add(id(par))
                    par_new_name = par_name
                    if cell_name:
                        par_new_name = cell_name + '.' + par_new_name