From 3a94d49214cd62db5f544974d7d941f650562ea2 Mon Sep 17 00:00:00 2001
From: Vincent TEMPLIER <vincent.templier@cea.fr>
Date: Thu, 26 Oct 2023 14:57:11 +0000
Subject: [PATCH] Fix missing outputs dims for last layer
---
aidge_core/export/node_export.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/aidge_core/export/node_export.py b/aidge_core/export/node_export.py
index 32629ae51..980cb05a5 100644
--- a/aidge_core/export/node_export.py
+++ b/aidge_core/export/node_export.py
@@ -43,7 +43,9 @@ class ExportNode(ABC):
for idx, child_node in enumerate(self.node.get_children()):
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
def export(self, export_folder:str, list_configs:list):
--
GitLab