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
