add fallback to optim.lr in AnnealingLearner

2017-09-25 06:10:54 +00:00 · 2017-09-25 06:10:54 +00:00 · a760c4841b
parent 916c6fe1f0
commit a760c4841b
1 changed files with 1 additions and 1 deletions
--- a/onn_core.py
+++ b/onn_core.py
@ -1271,7 +1271,7 @@ class AnnealingLearner(Learner):
        super().__init__(optim, epochs, rate)

    def rate_at(self, epoch):
-        return self.start_rate * self.anneal**epoch
+        return super().rate_at(epoch) * self.anneal**epoch

 def cosmod(x):
    # plot: https://www.desmos.com/calculator/hlgqmyswy2