github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/trainer/hooks
History
Liang Bowen ec5086c49c Refactored docstring to google style 2022-03-29 17:17:47 +08:00
..
__init__.py Layer integration (#83) 2021-12-27 15:04:32 +08:00
_base_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_checkpoint_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_log_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_lr_scheduler_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_mem_tracer_hook.py [profiler] add MemProfiler (#356) 2022-03-29 12:48:34 +08:00
_metric_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00