torch.mtia.memory.reset_peak_memory_stats

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

Сбрасывает статистику пикового использования памяти для указанного устройства.

Параметры:

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