Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[zero] initialize a stateful tensor manager #614

Merged
merged 6 commits into from
Apr 6, 2022
Merged

[zero] initialize a stateful tensor manager #614

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
5525436
[zero] initialize a stateful tensor manager
feifeibear Apr 1, 2022
538cf03
polish
feifeibear Apr 1, 2022
640be94
polish code
feifeibear Apr 1, 2022
eda1305
polish
feifeibear Apr 1, 2022
d8c5df4
polish
feifeibear Apr 1, 2022
2a5e23d
Merge branch 'main' of github.com:hpcaitech/ColossalAI into zero/stat…
feifeibear Apr 6, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
polish
  • Loading branch information
@feifeibear
feifeibear committed Apr 1, 2022
commit eda13056e25e98822a816ec011b281cea1678c09
2 changes: 1 addition & 1 deletion colossalai/utils/memory_tracer/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from .async_memtracer import AsyncMemoryMonitor
from .memstats_collector import MemStatsCollector

__all__ = ['AsyncMemoryMonitor', 'ModelDataTracer']
__all__ = ['AsyncMemoryMonitor', 'MemStatsCollector']