diff --git a/composer/loggers/mlflow_logger.py b/composer/loggers/mlflow_logger.py index 7a8acdd511..eb2f917a92 100644 --- a/composer/loggers/mlflow_logger.py +++ b/composer/loggers/mlflow_logger.py @@ -601,7 +601,7 @@ def log_images( assert isinstance(self._run_id, str) self._mlflow_client.log_image( image=image, - key=f'{name}_{step}_{im_ind}', + key=f'{name}_{im_ind}', run_id=self._run_id, step=step, )