notwa/optim
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

add a convenient generator method to Learner

Browse source
This commit is contained in:
Connor Olding 2019-02-18 06:23:05 +01:00
parent 2b5798332d
commit 69e6ec3fad
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onn/learner.py
View file

@ -50,13 +50,17 @@ class Learner:
return False return False
return True return True
def batch(self, progress): # TODO: rename def batch(self, progress): # TODO: rename.
# interpolates rates between epochs. # interpolates rates between epochs.
# unlike epochs, we do not store batch number as a state. # unlike epochs, we do not store batch number as a state.
# i.e. calling next() will not respect progress. # i.e. calling next() will not respect progress.
assert 0 <= progress <= 1 assert 0 <= progress <= 1
self.rate = self.rate_at(self._epoch + progress) self.rate = self.rate_at(self._epoch + progress)
def each_epoch(self): # TODO: rename?
while self.next():
yield self.epoch
@property @property
def final_rate(self): def final_rate(self):
return self.rate_at(self.epochs - 1e-8) return self.rate_at(self.epochs - 1e-8)