diff --git a/colossalai/trainer/hooks/_lr_scheduler_hook.py b/colossalai/trainer/hooks/_lr_scheduler_hook.py index 6f39e4400..c6da33442 100644 --- a/colossalai/trainer/hooks/_lr_scheduler_hook.py +++ b/colossalai/trainer/hooks/_lr_scheduler_hook.py @@ -18,6 +18,7 @@ class LRSchedulerHook(MetricHook): defaults to 1. If different hooks share same priority, the order of printing would depend on the hooks order in the hook list. """ + def __init__( self, lr_scheduler,