From 1adad3ece601a324c14c0ff2113165364a8d7c92 Mon Sep 17 00:00:00 2001 From: Colombo Date: Wed, 18 Nov 2020 13:38:27 +0400 Subject: [PATCH] fix RMSprop.py --- 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 37cebe9..70fd1be 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.lr, self.rho, self.epsilon, self.iterations] + list(self.accumulators_dict.values()) + return [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