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
def __init__(self, step_rate): super(SGD_origin, self).__init__() self.step_rate = shared_scalar(step_rate, name='step_rate')