torch.xpu.memory.reset_peak_memory_stats

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

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

Подробнее см. memory_stats(). Статистика пиковых значений соответствует ключу “peak” в каждом отдельном словаре статистики.

Параметры:

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