Browse Source

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

tags/v1.0.0
w30006229 5 years ago
parent
commit
4f0c7a8c45
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/nn/cell.py

+ 2
- 2
mindspore/nn/cell.py View File

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


Loading…
Cancel
Save