Code
Hub
Workspaces
Connect
Indexed graphs
Engine
MCP
copy
hub
/
github.com/InternLM/InternLM
/ simple_memory_profiler.py
File
simple_memory_profiler.py
internlm/utils/simple_memory_profiler.py:None–None ·
view source on GitHub ↗
Source
from the content-addressed store, hash-verified
1
import
os
2
import
time
3
from
collections
import
OrderedDict
4
from
functools
import
partial, reduce
Callers
nothing calls this directly
Calls
6
SimpleModel
Class · 0.85
SimpleMemoryProfiler
Class · 0.85
_simple_schedule
Function · 0.85
zero_grad
Method · 0.45
backward
Method · 0.45
step
Method · 0.45
Tested by
no test coverage detected