Remove step in log_image for MLFlow (#3601)

mosaicml · Sep 4, 2024 · 7d34d66 · 7d34d66
1 parent 998198d
commit 7d34d66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
                 )