Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/torch-spline-conv.git" did not exist on "26c0b8667f7be9ba948251f3c604d2cb5b1aed4e"
Unverified Commit c0b93a1c authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

correct the correction (#6163)

parent a2f6d521
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/benchmark/benchmark_utils.py
View file @ c0b93a1c
...@@ -310,7 +310,7 @@ def start_memory_tracing( ...@@ -310,7 +310,7 @@ def start_memory_tracing(
gpus_to_trace: Optional[List[int]] = None, gpus_to_trace: Optional[List[int]] = None,
) -> MemoryTrace: ) -> MemoryTrace:
""" Setup line-by-line tracing to record rss mem (RAM) at each line of a module or sub-module. """ Setup line-by-line tracing to record rss mem (RAM) at each line of a module or sub-module.
See `../../../examples/benchmarking/` for usage examples. See `./benchmark.py` for usage examples.
Current memory consumption is returned using psutil and in particular is the RSS memory Current memory consumption is returned using psutil and in particular is the RSS memory
"Resident Set Size” (the non-swapped physical memory the process is using). "Resident Set Size” (the non-swapped physical memory the process is using).
See https://psutil.readthedocs.io/en/latest/#psutil.Process.memory_info See https://psutil.readthedocs.io/en/latest/#psutil.Process.memory_info
......
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