From 9dd04709b7d662c5687c6863c762759901c49acf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9goire=20KUBLER?= <gregoire.kubler@proton.me>
Date: Wed, 6 Nov 2024 17:16:49 +0100
Subject: [PATCH] fix : add operator nb input argument was removed
---
src/loss/classification/BCE.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/src/loss/classification/BCE.cpp b/src/loss/classification/BCE.cpp
index 4ef1d8f..857bfe7 100644
--- a/src/loss/classification/BCE.cpp
+++ b/src/loss/classification/BCE.cpp
@@ -60,8 +60,8 @@ Aidge::Tensor Aidge::loss::BCE(std::shared_ptr<Tensor>& prediction,
const std::shared_ptr<Node> target_node = Producer(target, "label");
// Define nodes: add1 = prediction + eps1, add2 = target + eps1
- const std::shared_ptr<Node> add1_node = Add(2, "add1");
- const std::shared_ptr<Node> add2_node = Add(2, "add2");
+ const std::shared_ptr<Node> add1_node = Add("add1");
+ const std::shared_ptr<Node> add2_node = Add("add2");
prediction_node->addChild(add1_node, 0, 0);
Producer(std::make_shared<Tensor>(Array1D<float, 1>{{eps1}}))
->addChild(add1_node, 0, 1);
@@ -107,7 +107,7 @@ Aidge::Tensor Aidge::loss::BCE(std::shared_ptr<Tensor>& prediction,
sub1_node->addChild(div2_node, 0, 1);
// Define node: add3 = (target + eps1) / (prediction + eps1) - (1 - target + eps2)/(1 - prediction + eps2)
- const std::shared_ptr<Node> add3_node = Add(2, "add3");
+ const std::shared_ptr<Node> add3_node = Add("add3");
div1_node->addChild(add3_node, 0, 0);
div2_node->addChild(add3_node, 0, 1);
--
GitLab