verl/utils/debug/performance.py

• log_gpu_memory_usage · function · L20-L30 — def log_gpu_memory_usage(head: str, logger: logging.Logger = None, level=logging.DEBUG, rank: int = 0)