ver217
|
232142f402
|
[utils] refactor profiler (#837)
* add model data profiler
* add a subclass of torch.profiler.profile
* refactor folder structure
* remove redundant codes
* polish code
* use GeminiMemoryManager
* fix import path
* fix stm profiler ext
* polish comments
* remove useless file
|
2022-04-24 17:03:59 +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 |
|
HELSON
|
1ed7c24c02
|
Added PCIE profiler to dectect data transmission (#373)
|
2022-03-11 15:50:28 +08:00 |
|
HELSON
|
425bb0df3f
|
Added Profiler Context to manage all profilers (#340)
|
2022-03-11 15:50:28 +08:00 |
|
1SAA
|
73bff11288
|
Added profiler communication operations
Fixed bug for learning rate scheduler
|
2022-03-11 15:50:28 +08:00 |