Skip to content
Snippets Groups Projects
Commit 3a94d492 authored by Vincent Templier's avatar Vincent Templier
Browse files

Fix missing outputs dims for last layer

parent 491c4409
No related branches found
No related tags found
1 merge request!43[NodeExport] Add node export class
Pipeline #33430 passed
...@@ -43,7 +43,9 @@ class ExportNode(ABC): ...@@ -43,7 +43,9 @@ class ExportNode(ABC):
for idx, child_node in enumerate(self.node.get_children()): for idx, child_node in enumerate(self.node.get_children()):
self.outputs.append(child_node) self.outputs.append(child_node)
self.outputs_dims.append(self.operator.output(idx).dims())
# Dirty hot fix, change it quickly
self.outputs_dims.append(self.operator.output(0).dims())
@abstractmethod @abstractmethod
def export(self, export_folder:str, list_configs:list): def export(self, export_folder:str, list_configs:list):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment