def log_scores(self, tag, scores: dict, step=None):
    step = (step or self.global_step)
    for (k, v) in scores.items():
        self.log(((tag + '/') + k), v)
    mlflow.log_metrics({((tag + '/') + k): v.item() for (k, v) in scores.items()}, step=step)
def log_scores(self, tag, scores: dict, step=None):
    step = (step or self.global_step)
    for (k, v) in scores.items():
        self.log(((tag + '/') + k), v)
    mlflow.log_metrics({((tag + '/') + k): v.item() for (k, v) in scores.items()}, step=step)