def on_epoch_end(self, epoch, logs=None):
    if (not logs):
        return
    try_mlflow_log(mlflow.log_metrics, logs, step=epoch)
def on_epoch_end(self, epoch, logs=None):
    if (not logs):
        return
    try_mlflow_log(mlflow.log_metrics, logs, step=epoch)