torch.cuda.memory.reset_peak_host_memory_stats

torch.cuda.memory.reset_peak_host_memory_stats()

重置主機記憶體分配器跟蹤的“峰值”統計資訊。

有關詳細資訊，請參閱 host_memory_stats()。峰值統計資訊對應於每個單獨統計字典中的 “peak” 鍵。