DeepSpeed icon indicating copy to clipboard operation
DeepSpeed copied to clipboard

Published 20 hours ago • verified publisher icon microsoft

deepspeed/runtime/utils.py: reset_peak_memory_stats when empty cache

Open guoyejun opened this issue 2 years ago • 1 comments

guoyejun avatar Feb 08 '23 02:02 guoyejun

to make the logic clear. The peek stats should be cleared after cache is empty.

guoyejun avatar Feb 08 '23 02:02 guoyejun