From 874a7eba18f21f971db699fc3066daabd87ab973 Mon Sep 17 00:00:00 2001 From: Colombo Date: Wed, 18 Nov 2020 13:41:19 +0400 Subject: [PATCH] 1 --- core/leras/optimizers/RMSprop.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/leras/optimizers/RMSprop.py b/core/leras/optimizers/RMSprop.py index 70fd1be..37cebe9 100644 --- a/core/leras/optimizers/RMSprop.py +++ b/core/leras/optimizers/RMSprop.py @@ -23,7 +23,7 @@ class RMSprop(nn.OptimizerBase): self.lr_rnds_dict = {} def get_weights(self): - return [self.iterations] + list(self.accumulators_dict.values()) + return [self.lr, self.rho, self.epsilon, self.iterations] + list(self.accumulators_dict.values()) def initialize_variables(self, trainable_weights, vars_on_cpu=True, lr_dropout_on_cpu=False): # Initialize here all trainable variables used in training