Skip to content

Commit c2e29a3

Browse files
committed
enable system recording
1 parent fb67735 commit c2e29a3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

trinity/utils/monitor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414

1515
try:
1616
import mlflow
17+
18+
mlflow.enable_system_metrics_logging()
1719
except ImportError:
1820
mlflow = None
1921
from torch.utils.tensorboard import SummaryWriter

0 commit comments

Comments
 (0)