MCPcopy
hub / github.com/huggingface/transformers / build_cpu_memory_monitor

Function build_cpu_memory_monitor

src/transformers/testing_utils.py:4410–4425  ·  view source on GitHub ↗

Build and initialize a CPU memory monitor. Args: logger_instance: Optional logger to log initialization info. If None, no logging is done. Returns: CPUMemoryMonitor instance.

(logger_instance: logging.Logger | None = None)

Source from the content-addressed store, hash-verified

4408
4409
4410def build_cpu_memory_monitor(logger_instance: logging.Logger | None = None) -> CPUMemoryMonitor:
4411 """Build and initialize a CPU memory monitor.
4412
4413 Args:
4414 logger_instance: Optional logger to log initialization info. If None, no logging is done.
4415
4416 Returns:
4417 CPUMemoryMonitor instance.
4418 """
4419 monitor = CPUMemoryMonitor()
4420 if logger_instance is not None:
4421 if is_psutil_available():
4422 logger_instance.info(f"CPU memory monitor initialized: {monitor.total_memory_gib:.2f} GiB total")
4423 else:
4424 logger_instance.warning("psutil not available, memory monitoring disabled")
4425 return monitor
4426
4427
4428def convert_all_safetensors_to_bins(folder: str):

Callers 1

test_training_overfitMethod · 0.90

Calls 4

CPUMemoryMonitorClass · 0.85
is_psutil_availableFunction · 0.85
warningMethod · 0.80
infoMethod · 0.45

Tested by

no test coverage detected