Skip to content

Commit 7dfd4d3

Browse files
Reduce system metrics logging frequency (#3604)
1 parent 7d34d66 commit 7dfd4d3

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

composer/loggers/mlflow_logger.py

+8-4
Original file line numberDiff line numberDiff line change
@@ -174,9 +174,9 @@ def __init__(
174174

175175
if log_system_metrics:
176176
# Set system metrics sampling interval and samples before logging so that system metrics
177-
# are collected every 5s, and aggregated over 3 samples before being logged
178-
# (logging per 15s).
179-
mlflow.set_system_metrics_samples_before_logging(3)
177+
# are collected every 5s, and aggregated over 6 samples before being logged
178+
# (logging per 30s).
179+
mlflow.set_system_metrics_samples_before_logging(6)
180180
mlflow.set_system_metrics_sampling_interval(5)
181181

182182
self._rank_zero_only = rank_zero_only
@@ -545,7 +545,11 @@ def register_model_with_run_id(
545545
"""
546546
if self._enabled:
547547
from mlflow.exceptions import MlflowException
548-
from mlflow.protos.databricks_pb2 import ALREADY_EXISTS, RESOURCE_ALREADY_EXISTS, ErrorCode
548+
from mlflow.protos.databricks_pb2 import (
549+
ALREADY_EXISTS,
550+
RESOURCE_ALREADY_EXISTS,
551+
ErrorCode,
552+
)
549553

550554
full_name = f'{self.model_registry_prefix}.{name}' if len(self.model_registry_prefix) > 0 else name
551555

0 commit comments

Comments
 (0)