Houssem ROUIS
--- a/include/aidge/learning/optimizer/SGD.hpp
+++ b/include/aidge/learning/optimizer/SGD.hpp
        mGradientInertia = std::vector<Tensor>(parameters.size());
        for (std::size_t i = 0; i < parameters.size(); ++i) {
            mGradientInertia[i] = Tensor(parameters[i]->dims());
+            mGradientInertia[i].setBackend(parameters[i]->backend());
+        }
+        if (parameters.size() > 0) {
+            mReversedDampening.setBackend(mParameters[0]->getImpl()->backend());
+            mMomentum.setBackend(mParameters[0]->getImpl()->backend());