torch.cuda.reset_peak_host_memory_stats
- torch.cuda.reset_peak_host_memory_stats()[source][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.