Unfix multi outputs

CEA-LIST · Jun 27, 2024 · bc96033 · bc96033
1 parent ece4517
commit bc96033
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/n2d2/deepnet.py b/python/n2d2/deepnet.py
@@ -111,10 +111,16 @@ def get_output_cells(self):
         Return the last N2D2 cells in the deepNet
         """
         output = []
+        """ 
+        // Made to support several outputs but breaks regular networks exports
         for cell in self.N2D2().getCells():
             if len(self._cells[cell].N2D2().getChildrenCells()) == 0:
                 output.append(self._cells[cell])
         return output
+        """
+        for cell in self.N2D2().getLayers()[-1]:
+            output.append(self._cells[cell])
+        return output
 
     def draw(self, filename):
         N2D2.DrawNet.draw(self._N2D2_object, filename)