add log_model to MLFlowParams (#7258)

Signed-off-by: Alireza Morsali <[email protected]>
NVIDIA · Aug 23, 2023 · d091675 · d091675
1 parent 088df45
commit d091675
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nemo/utils/loggers/mlflow_logger.py b/nemo/utils/loggers/mlflow_logger.py
@@ -29,3 +29,5 @@ class MLFlowParams:
     artifact_location: Optional[str] = None
     # provide run_id if resuming a previously started run
     run_id: Optional[str] = None
+    # Log checkpoints created by ModelCheckpoint as MLFlow artifacts.
+    log_model: bool = False