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
Jiarui Fang 8c66a1d0aa
[polish] remove useless file _mem_tracer_hook.py (#1963)
2 years ago
..
__init__.py
_base_hook.py
_checkpoint_hook.py [NFC] polish _checkpoint_hook.py code style (#1722) 2 years ago
_commons_.py
_log_hook.py
_lr_scheduler_hook.py
_metric_hook.py