log_memory_stats

torchtune.training.log_memory_stats(stats: Dict[str, float], message: str = '模型初始化后的内存统计信息：') → None

将包含内存统计信息的字典记录到日志记录器。stats 应包含字段 peak_memory_active、peak_memory_alloc 和 peak_memory_reserved，这些字段由 torchtune.training.get_memory_stats() 返回。

参数:

• stats (Dict[str, float]) – 包含峰值内存活动、峰值内存分配和峰值内存保留统计信息的字典。

• message (str) – 要添加到日志输出的可选消息。默认为 “模型初始化后的内存统计信息：”