utils.py: reset_peak_memory_stats when empty cache (#2803)

Commit

2 years ago

deepspeed/runtime/utils.py: reset_peak_memory_stats when empty cache (#2803) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

References

Author

guoyejun

Parents