diff --git a/trinity/utils/monitor.py b/trinity/utils/monitor.py index f0e847f4d2..0eee105608 100644 --- a/trinity/utils/monitor.py +++ b/trinity/utils/monitor.py @@ -194,6 +194,7 @@ def __init__( os.environ["MLFLOW_TRACKING_PASSWORD"] = password mlflow.set_tracking_uri(config.monitor.monitor_args.get("uri", "http://localhost:5000")) mlflow.set_experiment(project) + mlflow.enable_system_metrics_logging() mlflow.start_run( run_name=f"{name}_{role}", tags={