Skip to content
Snippets Groups Projects

Fix loss derivative

Merged Olivier Antoni requested to merge oantoni/aidge_learning:Fix_loss_derivative into dev
2 files
+ 2
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -123,7 +123,7 @@ Aidge::Tensor Aidge::loss::BCE(std::shared_ptr<Tensor>& prediction,
// Define node: gradient
const std::shared_ptr<Node> gradient_node = Mul("gradient");
div1_node->addChild(gradient_node, 0, 0);
Producer(std::make_shared<Tensor>(Array1D<float, 1>{{-1.0f/float(target->dims()[0])}}))
Producer(std::make_shared<Tensor>(Array1D<float, 1>{{-1.0f/float(target->size())}}))
->addChild(gradient_node, 0, 1);
// Create GraphView
Loading