def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, *args, **kwargs): #self.__dict__.update(locals()) self.iterations = shared_scalar(0)
def __init__(self, momentum_schedule=[(0, 0.)], momentum_var=shared_scalar(0.)): super(MomentumScheduler, self).__init__() self.momentum_schedule = momentum_schedule self.momentum_var = momentum_var
def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, *args, **kwargs): super(CustomSGD, self).__init__(**kwargs) self.__dict__.update(locals()) self.iterations = shared_scalar(0) self.lr = shared_scalar(lr) self.momentum = shared_scalar(momentum)