Browse Source

!1625 fix SequentialCell

Merge pull request !1625 from liuxiao78/master
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
0b4de00176
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/nn/layer/container.py

+ 3
- 0
mindspore/nn/layer/container.py View File

@@ -105,6 +105,9 @@ class SequentialCell(Cell):
self.insert_child_to_cell(name, cell)
else:
raise TypeError('Cells must be list or orderedDict')
else:
for index, cell in enumerate(args):
self.insert_child_to_cell(str(index), cell)
self.cell_list = list(self._cells.values())

def __getitem__(self, index):


Loading…
Cancel
Save