From 3055399d7d87b05d19a16675d63803fbf6f48c25 Mon Sep 17 00:00:00 2001
From: Jerome Hue <jerome.hue@cea.fr>
Date: Thu, 5 Dec 2024 19:08:38 +0100
Subject: [PATCH] chore: fix leaky output
---
src/operator/MetaOperatorDefs/Leaky.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/operator/MetaOperatorDefs/Leaky.cpp b/src/operator/MetaOperatorDefs/Leaky.cpp
index 450f2d1c3..425d47a49 100644
--- a/src/operator/MetaOperatorDefs/Leaky.cpp
+++ b/src/operator/MetaOperatorDefs/Leaky.cpp
@@ -65,7 +65,7 @@ std::shared_ptr<Node> Leaky(const int nbTimeSteps, const float threshold, const
addNode->addChild(subNode, 0, 0);
reset->addChild(subNode, 0, 1);
- // U[t] = Input[T] + beta * U[T-1] - S[T-1]
+ // U[t] = (Input[T] + beta * U[T-1]) - S[T-1]
subNode->addChild(potentialMem, 0, 0);
// U[T] - U_th
@@ -96,7 +96,7 @@ std::shared_ptr<Node> Leaky(const int nbTimeSteps, const float threshold, const
// This way, we can plug a stack operator after or node, and get correct results
microGraph->setOrderedOutputs({//{potentialMem, memorizeOpDataOutputIndex},
//{spikeMem, memorizeOpDataOutputIndex}
- {addNode, 0},
+ {subNode, 0},
{hsNode, 0}});
auto metaOp = MetaOperator(/*type*/ "Leaky",
--
GitLab