ColossalAI

Commit Graph

Author	SHA1	Message	Date
Jiarui Fang	4d9332b4c5	[refactor] moving memtracer to gemini (#801 )	2022-04-19 10:13:08 +08:00
HELSON	340e59f968	[utils] add synchronized cuda memory monitor (#740 )	2022-04-13 10:50:54 +08:00
Jie Zhu	73d36618a6	[profiler] add MemProfiler (#356 ) * add memory trainer hook * fix bug * add memory trainer hook * fix import bug * fix import bug * add trainer hook * fix #370 git log bug * modify `to_tensorboard` function to support better output * remove useless output * change the name of `MemProfiler` * complete memory profiler * replace error with warning * finish trainer hook * modify interface of MemProfiler * modify `__init__.py` in profiler * remove unnecessary pass statement * add usage to doc string * add usage to trainer hook * new location to store temp data file	2022-03-29 12:48:34 +08:00

Author

SHA1

Message

Date

Jiarui Fang

4d9332b4c5

[refactor] moving memtracer to gemini (#801 )

2022-04-19 10:13:08 +08:00

HELSON

340e59f968

[utils] add synchronized cuda memory monitor (#740 )

2022-04-13 10:50:54 +08:00

Jie Zhu

73d36618a6

[profiler] add MemProfiler (#356 )

* add memory trainer hook

* fix bug

* add memory trainer hook

* fix import bug

* fix import bug

* add trainer hook

* fix #370 git log bug

* modify `to_tensorboard` function to support better output

* remove useless output

* change the name of `MemProfiler`

* complete memory profiler

* replace error with warning

* finish trainer hook

* modify interface of MemProfiler

* modify `__init__.py` in profiler

* remove unnecessary pass statement

* add usage to doc string

* add usage to trainer hook

* new location to store temp data file

2022-03-29 12:48:34 +08:00

3 Commits (193d6293112d3c36af923a076086989cb631b7fd)