remove unused attribute lr_scheduler_interval

finegrain-ai · Dec 20, 2024 · 9e49bc5 · 9e49bc5
1 parent 51bbd4f
commit 9e49bc5
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 4 deletions.
diff --git a/src/refiners/training_utils/clock.py b/src/refiners/training_utils/clock.py
@@ -22,12 +22,10 @@ def __init__(
         self,
         training_duration: TimeValue,
         gradient_accumulation: Step,
-        lr_scheduler_interval: Iteration | Epoch,
         verbose: bool = True,
     ) -> None:
         self.training_duration = training_duration
         self.gradient_accumulation = gradient_accumulation
-        self.lr_scheduler_interval = lr_scheduler_interval
         self.verbose = verbose
         self.start_time = None
         self.end_time = None

diff --git a/src/refiners/training_utils/trainer.py b/src/refiners/training_utils/trainer.py
@@ -139,7 +139,6 @@ def clock(self, config: ClockConfig) -> TrainingClock:
         return TrainingClock(
             training_duration=self.config.training.duration,
             gradient_accumulation=self.config.training.gradient_accumulation,
-            lr_scheduler_interval=self.config.lr_scheduler.update_interval,
             verbose=config.verbose,
         )
 

diff --git a/tests/training_utils/test_trainer.py b/tests/training_utils/test_trainer.py
@@ -205,7 +205,6 @@ def training_clock() -> TrainingClock:
     return TrainingClock(
         training_duration=Epoch(5),
         gradient_accumulation=Step(1),
-        lr_scheduler_interval=Epoch(1),
     )