Skip to content

Commit 3343729

Browse files
authored
Enable System Metrics Recording in MLFLOW (#362)
1 parent fb67735 commit 3343729

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

trinity/utils/monitor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,7 @@ def __init__(
194194
os.environ["MLFLOW_TRACKING_PASSWORD"] = password
195195
mlflow.set_tracking_uri(config.monitor.monitor_args.get("uri", "http://localhost:5000"))
196196
mlflow.set_experiment(project)
197+
mlflow.enable_system_metrics_logging()
197198
mlflow.start_run(
198199
run_name=f"{name}_{role}",
199200
tags={

0 commit comments

Comments
 (0)