This commit is contained in:
Patrick von Platen 2020-06-11 21:28:01 +02:00 коммит произвёл GitHub
Родитель 6293eb04df
Коммит caf3746678
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 5 добавлений и 6 удалений

Просмотреть файл

@ -129,7 +129,6 @@ class PyTorchBenchmark(Benchmark):
logger.info(
"Please consider updating PyTorch to version 1.4 to get more accuracy on GPU memory usage"
)
memory = Memory(torch.cuda.max_memory_cached())
memory = Memory(torch.cuda.max_memory_reserved())
return memory, summary