!1625 fix SequentialCell

Merge pull request !1625 from liuxiao78/master
2020-05-29 10:16:06 +08:00 · 2020-05-29 10:16:06 +08:00 · 0b4de00176
parent bb72c0020b 90b51d6c89
commit 0b4de00176
1 changed files with 3 additions and 0 deletions
--- a/mindspore/nn/layer/container.py
+++ b/mindspore/nn/layer/container.py
@ -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):