https://pytorch.org/docs/master/cuda.html : Both reset_peak_stats and reset_peak_memory_stats are mentioned, but no docs are available. cc @jlin27