torch.accelerator.memory.reset_accumulated_memory_stats

torch.accelerator.memory.reset_accumulated_memory_stats(device_index=None, /) [источник]

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

Параметры:

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

Примечание

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