Skip to content
Snippets Groups Projects
Commit 44b250cd authored by Benjamin Halimi's avatar Benjamin Halimi Committed by Benjamin Halimi
Browse files

Apply 1 suggestion(s) to 1 file(s)


Co-authored-by: default avatarMaxence Naud <maxence.naud@cea.fr>
parent ab69998e
No related branches found
No related tags found
2 merge requests!44Update 0.2.3 -> 0.3.0,!32Add the Cross Entropy Loss
......@@ -30,7 +30,6 @@
#include "aidge/operator/ReduceSum.hpp"
#include "aidge/operator/Softmax.hpp"
#include "aidge/graph/OpArgs.hpp"
Aidge::Tensor Aidge::loss::CELoss(std::shared_ptr<Tensor>& prediction,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment