Fixed docstring in colossalai (#171)

2025-09-07 20:10:17 +00:00 · 2022-01-21 10:44:30 +08:00
parent e2089c5c15
commit 0f8c7f9804
77 changed files with 983 additions and 603 deletions
--- a/colossalai/trainer/hooks/_lr_scheduler_hook.py
+++ b/colossalai/trainer/hooks/_lr_scheduler_hook.py
@@ -15,7 +15,6 @@ class LRSchedulerHook(MetricHook):
    :type store_lr_in_state: bool, optional
    :param priority: Priority in the printing, hooks with small priority will be printed in front, defaults to 1
    :type priority: int, optional
-    :param trainer: Trainer attached with current hook
    """
    def __init__(
        self,