fix SequentialCell

2020-05-28 22:37:06 +08:00 · 2020-05-28 22:37:06 +08:00 · 90b51d6c89
parent 21da86b393
commit 90b51d6c89
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):