def get_lr_metric(optimizer):
def lr(y_true, y_pred):
return optimizer.lr
return lr
lr_metric = get_lr_metric(optimizer)
model.compile(loss='binary_crossentropy', optimizer=optimizer, metrics=['acc', lr_metric])
参考https://stackoverflow.com/questions/48198031/keras-add-variables-to-progress-bar/48206009#48206009
网友评论