torch.cuda.memory.reset_peak_host_memory_stats

torch.cuda.memory.reset_peak_host_memory_stats() [исходный код]

Сбрасывает отслеживаемую аллокатором памяти хоста «пиковую» статистику.

Подробности см. в host_memory_stats(). Пиковая статистика соответствует ключу “peak” в каждом отдельном словаре статистики.