Skip to content
Snippets Groups Projects
Commit 5d8c7cd1 authored by Lorenzo Moneta's avatar Lorenzo Moneta
Browse files

Use the template type (Scalar_t) instead of double, otherwise it will not...

Use the template type (Scalar_t) instead of double, otherwise it will not compile for networks based on float types
parent e2545b4e
No related branches found
No related tags found
No related merge requests found
...@@ -208,7 +208,7 @@ auto testOptimization(typename Architecture_t::Scalar_t momentum, EOptimizer opt ...@@ -208,7 +208,7 @@ auto testOptimization(typename Architecture_t::Scalar_t momentum, EOptimizer opt
deepNet.Forward(I, false); deepNet.Forward(I, false);
// get the output of the last layer of the deepNet // get the output of the last layer of the deepNet
TMatrixT<Double_t> Ytemp(deepNet.GetLayerAt(deepNet.GetLayers().size() - 1)->GetOutputAt(0)); TMatrixT<Scalar_t> Ytemp(deepNet.GetLayerAt(deepNet.GetLayers().size() - 1)->GetOutputAt(0));
std::cout << " Before Training: Mean Absolute Error = " << meanAbsoluteError(Ytemp, K) << ","; std::cout << " Before Training: Mean Absolute Error = " << meanAbsoluteError(Ytemp, K) << ",";
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment