def get_optimizer(config_data):
options = config_data['optimizer']
name = options['name']
if name == 'adadelta':
return optimizers.Adadelta(lr=options['lr'], rho=options['rho'], epsilon=options['epsilon'])
else:
return optimizers.SGD()
评论列表
文章目录