From 0ec99fd5c0f2e3e7d6d982c2c9362cc6333626c7 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Thu, 28 Mar 2024 14:57:20 +0000
Subject: [PATCH] upd SGD
---
include/aidge/learning/optimizer/SGD.hpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/aidge/learning/optimizer/SGD.hpp b/include/aidge/learning/optimizer/SGD.hpp
index 854918a..2f77281 100644
--- a/include/aidge/learning/optimizer/SGD.hpp
+++ b/include/aidge/learning/optimizer/SGD.hpp
@@ -52,7 +52,7 @@ public:
mReversedDampening.set<float>(0, 1.0f - dampening);
}
- void update() override {
+ void update() override final {
mLR.setBackend(mParameters[0]->getImpl()->backend());
mLR.set<float>(0, learningRate());
if (mParameters[0]->getImpl()->backend() != mMomentum.getImpl()->backend()) {
@@ -74,7 +74,7 @@ public:
mLRScheduler.update();
}
- void setParameters(const std::vector<std::shared_ptr<Tensor>>& parameters) {
+ void setParameters(const std::vector<std::shared_ptr<Tensor>>& parameters) override final {
Optimizer::setParameters(parameters);
mGradientInertia = std::vector<Tensor>(parameters.size());
for (std::size_t i = 0; i < parameters.size(); ++i) {
--
GitLab