We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ece4517 commit bc96033Copy full SHA for bc96033
python/n2d2/deepnet.py
@@ -111,10 +111,16 @@ def get_output_cells(self):
111
Return the last N2D2 cells in the deepNet
112
"""
113
output = []
114
+ """
115
+ // Made to support several outputs but breaks regular networks exports
116
for cell in self.N2D2().getCells():
117
if len(self._cells[cell].N2D2().getChildrenCells()) == 0:
118
output.append(self._cells[cell])
119
return output
120
121
+ for cell in self.N2D2().getLayers()[-1]:
122
+ output.append(self._cells[cell])
123
+ return output
124
125
def draw(self, filename):
126
N2D2.DrawNet.draw(self._N2D2_object, filename)
0 commit comments