pytorch
84232b76 - docs: add `reset_peak_memory_stats` in cuda.rst (#54668)

Commit
3 years ago
docs: add `reset_peak_memory_stats` in cuda.rst (#54668) Summary: fixes https://github.com/pytorch/pytorch/issues/41808 https://11812999-65600975-gh.circle-artifacts.com/0/docs/cuda.html One question: does `reset_peak_stats` exist in `torch.cuda` ? I can't find anywhere. Pull Request resolved: https://github.com/pytorch/pytorch/pull/54668 Reviewed By: ailzhang Differential Revision: D27328444 Pulled By: zou3519 fbshipit-source-id: 098024d43da98e3249aa9aa71cb10126095504a4
Author
Jeff Yang
Parents
Loading