Skip to content

Commit bc96033

Browse files
author
Axel Farrugia
committed
Unfix multi outputs
1 parent ece4517 commit bc96033

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

python/n2d2/deepnet.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,10 +111,16 @@ def get_output_cells(self):
111111
Return the last N2D2 cells in the deepNet
112112
"""
113113
output = []
114+
"""
115+
// Made to support several outputs but breaks regular networks exports
114116
for cell in self.N2D2().getCells():
115117
if len(self._cells[cell].N2D2().getChildrenCells()) == 0:
116118
output.append(self._cells[cell])
117119
return output
120+
"""
121+
for cell in self.N2D2().getLayers()[-1]:
122+
output.append(self._cells[cell])
123+
return output
118124

119125
def draw(self, filename):
120126
N2D2.DrawNet.draw(self._N2D2_object, filename)

0 commit comments

Comments
 (0)