github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-16 16:32:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,466 Commits 56 Branches 51 Tags 236 MiB
feature/stable-diffusion
Commit Graph

2 Commits

Author SHA1 Message Date
digger yu
a9d1cadc49
fix typo with colossalai/trainer utils zero (#3908) 2023-06-07 16:08:37 +08:00
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