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