torch.cuda.memory.reset_peak_host_memory_stats

torch.cuda.memory.reset_peak_host_memory_stats() [source]

Reset the “peak” stats tracked by the host memory allocator.

See host_memory_stats() for details. Peak stats correspond to the “peak” key in each individual stat dict.