def __init__(self, lr, momentum):
     SGD.__init__(self, lr=lr, decay=1e-6, momentum=momentum, nesterov=True)
Esempio n. 2
0
 def __init__(self, **args):
     SGD.__init__(self, **args)
Esempio n. 3
0
 def __init__(self, lr, momentum):
     SGD.__init__(self, lr=lr, decay=1e-6, momentum=momentum, nesterov=True)