1

2020-11-18 13:41:19 +04:00
parent 1adad3ece6
commit 874a7eba18
1 changed files with 1 additions and 1 deletions
--- 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