diff --git a/src/loss/classification/CELoss.cpp b/src/loss/classification/CELoss.cpp index a892d44cefbd4370eaea25ae05db8f33ebe4437b..0a2ba760507dea492a09eceadafcd8910a2e4aa3 100644 --- a/src/loss/classification/CELoss.cpp +++ b/src/loss/classification/CELoss.cpp @@ -31,7 +31,6 @@ #include "aidge/operator/Softmax.hpp" - Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction, const std::shared_ptr<Tensor>& target) { @@ -121,5 +120,4 @@ Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction, // Return the loss value return (*lossTensor); - return (*lossTensor); } \ No newline at end of file