log_memory_stats

torchtune.training.log_memory_stats(stats: Dict[str, float]) → None

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

参数：

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