log_memory_stats¶ torchtune.training.log_memory_stats(stats: Dict[str, float], message: str = 'Memory stats after model init:') → None[source]¶ 将包含内存统计信息的字典记录到日志记录器中。stats 应包含字段 peak_memory_active、peak_memory_alloc 和 peak_memory_reserved，如 torchtune.training.get_memory_stats() 返回的那样。 Parameters: stats (Dict[str, float]) – 一个包含峰值内存活动、峰值内存分配和峰值内存保留统计信息的字典。 message (str) – 一个可选的消息，用于在日志输出前添加。 默认为“模型初始化后的内存统计：”
