torch.cuda.memory.reset_peak_memory_stats

torch.cuda.memory.reset_peak_memory_stats(device=None) [исходный код]

Сбросить статистику «пиковых» значений, отслеживаемую распределителем памяти CUDA.

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

Параметры:

device (torch.device или int, необязательно) – выбранное устройство. Возвращает статистику для текущего устройства, заданного current_device(), если device имеет значение None (по умолчанию).

Примечание

Подробнее об управлении памятью GPU см. в разделе Управление памятью.