Skip to content
Snippets Groups Projects
Commit 9f35ac3b authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

minor changes

parent 44b250cd
No related branches found
No related tags found
2 merge requests!44Update 0.2.3 -> 0.3.0,!32Add the Cross Entropy Loss
Hide whitespace changes
Inline Side-by-side
src/loss/classification/CELoss.cpp
+ 0
− 2
View file @ 9f35ac3b
...@@ -31,7 +31,6 @@ ...@@ -31,7 +31,6 @@
#include "aidge/operator/Softmax.hpp" #include "aidge/operator/Softmax.hpp"
Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction, Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction,
const std::shared_ptr<Tensor>& target) const std::shared_ptr<Tensor>& target)
{ {
...@@ -121,5 +120,4 @@ Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction, ...@@ -121,5 +120,4 @@ Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction,
// Return the loss value // Return the loss value
return (*lossTensor); return (*lossTensor);
return (*lossTensor);
} }
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent