ColossalAI/colossalai/trainer/hooks
1SAA 73bff11288 Added profiler communication operations
Fixed bug for learning rate scheduler
2022-03-11 15:50:28 +08:00
..
__init__.py Layer integration (#83) 2021-12-27 15:04:32 +08:00
_base_hook.py Fixed docstring in colossalai (#171) 2022-01-21 10:44:30 +08:00
_checkpoint_hook.py Fixed docstring in colossalai (#171) 2022-01-21 10:44:30 +08:00
_log_hook.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
_lr_scheduler_hook.py Added profiler communication operations 2022-03-11 15:50:28 +08:00
_metric_hook.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00