fix indentation issue (#4941)
This commit is contained in:
Родитель
6293eb04df
Коммит
caf3746678
|
@ -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
|
||||
|
|
Загрузка…
Ссылка в новой задаче