torch.accelerator.memory.reset_peak_memory_stats

torch.accelerator.memory.reset_peak_memory_stats(device_index=None, /) [source]

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

Параметры:

device_index (torch.device, str, int, optional) – индекс целевого устройства. Если не указан, по умолчанию используется torch.accelerator.current_device_index(). Если передан объект torch.device или строка, его тип должен совпадать с типом текущего устройства ускорителя.

Примечание

Эта функция ничего не делает, если аллокатор памяти текущего ускорителя не был инициализирован.