torch.mtia.memory.max_memory_allocated

torch.mtia.memory.max_memory_allocated(device=None) [source]

Возвращает максимальный объем памяти в байтах, выделенный для указанного устройства.

Параметры:

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

Тип возвращаемого значения:

int