Skip to content

GitLab

Unverified Commit caf37466 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix indentation issue (#4941)

parent 6293eb04
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 6 deletions
src/transformers/benchmark/benchmark.py
View file @ caf37466
...@@ -129,7 +129,6 @@ class PyTorchBenchmark(Benchmark): ...@@ -129,7 +129,6 @@ class PyTorchBenchmark(Benchmark):
logger.info( logger.info(
"Please consider updating PyTorch to version 1.4 to get more accuracy on GPU memory usage" "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()) memory = Memory(torch.cuda.max_memory_reserved())
return memory, summary return memory, summary
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment