torch.cuda.reset_peak_host_memory_stats

torch.cuda.reset_peak_host_memory_stats()

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

詳細資訊請參見 host_memory_stats()。峰值統計資料對應於每個獨立統計字典中的 “peak” 鍵。