tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

fix: cache process per comment

Browse Source
This commit is contained in:
NanoCode012
2025-10-31 14:57:07 +07:00
parent 6adb7c7fa3
commit c55742796a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/axolotl/telemetry/runtime_metrics.py
View File

@@ -118,6 +118,7 @@ class RuntimeMetricsTracker:
"""Initialize the runtime metrics tracker.""" """Initialize the runtime metrics tracker."""
self.metrics = RuntimeMetrics(start_time=time.time()) self.metrics = RuntimeMetrics(start_time=time.time())
self.telemetry_manager = TelemetryManager.get_instance() self.telemetry_manager = TelemetryManager.get_instance()
self._process = psutil.Process()
def start_epoch(self, epoch: int): def start_epoch(self, epoch: int):
"""Record the start of a new epoch.""" """Record the start of a new epoch."""
@@ -181,7 +182,7 @@ class RuntimeMetricsTracker:
def update_memory_metrics(self): def update_memory_metrics(self):
"""Update peak memory usage metrics.""" """Update peak memory usage metrics."""
# CPU memory # CPU memory
cpu_memory = psutil.Process().memory_info().rss cpu_memory = self._process.memory_info().rss
self.metrics.peak_cpu_memory = max(self.metrics.peak_cpu_memory, cpu_memory) self.metrics.peak_cpu_memory = max(self.metrics.peak_cpu_memory, cpu_memory)
# GPU memory (if available) # GPU memory (if available)
@@ -194,7 +195,7 @@ class RuntimeMetricsTracker:
def get_memory_metrics(self) -> dict[str, Any]: def get_memory_metrics(self) -> dict[str, Any]:
"""Get the current memory metrics as a dictionary.""" """Get the current memory metrics as a dictionary."""
memory_metrics = { memory_metrics = {
"cpu_memory_bytes": psutil.Process().memory_info().rss, "cpu_memory_bytes": self._process.memory_info().rss,
"peak_cpu_memory_bytes": self.metrics.peak_cpu_memory, "peak_cpu_memory_bytes": self.metrics.peak_cpu_memory,
} }