You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/trainer/hooks
LuGY 730f88f8e1
[NFC] polish _checkpoint_hook.py code style (#1722)
2 years ago
..
__init__.py [model checkpoint] updated checkpoint hook (#598) 3 years ago
_base_hook.py Refactored docstring to google style 3 years ago
_checkpoint_hook.py [NFC] polish _checkpoint_hook.py code style (#1722) 2 years ago
_commons_.py [refactor] moving memtracer to gemini (#801) 3 years ago
_log_hook.py fix issue #1080 (#1071) 3 years ago
_lr_scheduler_hook.py [NFC] polish ./colossalai/trainer/hooks/_lr_scheduler_hook.py code style (#1576) 2 years ago
_mem_tracer_hook.py [refactor] moving memtracer to gemini (#801) 3 years ago
_metric_hook.py fix issue #1080 (#1071) 3 years ago