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

nothing calls this directly

SimpleModelClass · 0.85

SimpleMemoryProfilerClass · 0.85

_simple_scheduleFunction · 0.85

zero_gradMethod · 0.45

backwardMethod · 0.45

stepMethod · 0.45

no test coverage detected