def get_optimizer(self, model, lr=1, momentum=0.9, weight_decay=0): optimizer = Utils.createoptimizer(model, lr=lr, momentum=momentum, weight_decay=weight_decay, nesterov=True) return optimizer
def get_optimizer(self, model, lr=1, momentum=0.9, weight_decay=0): ''' :param model: CNN model :param lr: learning rate :param momentum: momentum mostly used value is 0.9 :param weight_decay: weight decay or also known as l2 regulrization :return: optimizer object ''' optimizer = Utils.createoptimizer(model, lr=lr, momentum=momentum, weight_decay=weight_decay, nesterov=True) return optimizer