ColossalAI/colossalai/trainer/hooks
LuGY 80e37eec42
fix the ckpt bugs when using DDP (#769)
2022-04-14 21:03:24 +08:00
..
__init__.py [model checkpoint] updated checkpoint hook (#598) 2022-04-01 16:53:03 +08:00
_base_hook.py Refactored docstring to google style 2022-03-29 17:17:47 +08:00
_checkpoint_hook.py fix the ckpt bugs when using DDP (#769) 2022-04-14 21:03:24 +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 [utils] add synchronized cuda memory monitor (#740) 2022-04-13 10:50:54 +08:00
_metric_hook.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00