ColossalAI/colossalai/trainer/hooks
Jiarui Fang 8c66a1d0aa
[polish] remove useless file _mem_tracer_hook.py (#1963)
2022-11-16 15:55:10 +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 [NFC] polish _checkpoint_hook.py code style (#1722) 2022-10-19 12:20:51 +08:00
_commons_.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
_log_hook.py fix issue #1080 (#1071) 2022-06-07 17:21:11 +08:00
_lr_scheduler_hook.py [NFC] polish ./colossalai/trainer/hooks/_lr_scheduler_hook.py code style (#1576) 2022-09-08 22:11:04 +08:00
_metric_hook.py fix issue #1080 (#1071) 2022-06-07 17:21:11 +08:00