From 812878e6eb89dcb62bfc2ba7cdc16ba6f5cbce9a Mon Sep 17 00:00:00 2001 From: Falguni Ghosh <falguni.ghosh@fau.de> Date: Sun, 15 Oct 2023 21:00:58 +0000 Subject: [PATCH] Upload New File --- 1_DL_base/Optimizers.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 1_DL_base/Optimizers.py diff --git a/1_DL_base/Optimizers.py b/1_DL_base/Optimizers.py new file mode 100644 index 0000000..b778864 --- /dev/null +++ b/1_DL_base/Optimizers.py @@ -0,0 +1,14 @@ +class Sgd: + learning_rate = None + weight_tensor_updated = None + + def __init__(self, learning_rate): + self.learning_rate = learning_rate + + def calculate_update(self, weight_tensor, gradient_tensor): + self.weight_tensor_updated = weight_tensor - self.learning_rate * gradient_tensor + return self.weight_tensor_updated + + + + -- GitLab