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

allow integer expando values

Browse source
This commit is contained in:
Connor Olding 2017-06-18 01:58:34 +00:00
parent 35f380c5d0
commit 264c3abd83
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
optim_nn_core.py
View file

@ -1090,13 +1090,16 @@ class SGDR(Learner):
def __init__(self, optim, epochs=100, rate=None,
restarts=0, restart_decay=0.5, callback=None,
expando=None):
expando=0):
self.restart_epochs = int(epochs)
self.decay = _f(restart_decay)
self.restarts = int(restarts)
self.restart_callback = callback
# TODO: rename expando to something not insane
self.expando = expando if expando is not None else lambda i: i
if type(self.expando) == int:
inc = self.expando
self.expando = self.expando = lambda i: inc
self.splits = []
epochs = 0