Example #1
0
 def __init__(self, step_rate, decay, momentum, nesterov=False):
     super(SGD, self).__init__(step_rate)
     self.iterations = shared_scalar(0, name='iterations')
     self.decay = shared_scalar(decay, name='decay')
     self.momentum = shared_scalar(momentum, name='momentum')
     self.nesterov = nesterov
Example #2
0
 def __init__(self, step_rate):
     super(SGD_origin, self).__init__()
     self.step_rate = shared_scalar(step_rate, name='step_rate')