torch.cuda.memory.reset\_peak\_memory\_stats ============================================ .. currentmodule:: torch.cuda.memory .. autofunction:: reset_peak_memory_stats