torch.cuda.reset_peak_host_memory_stats

torch.cuda.reset_peak_host_memory_stats()

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.