Skip to content

Commit 4d75344

Browse files
committed
remove unused attribute lr_scheduler_interval
1 parent 50f865a commit 4d75344

File tree

3 files changed

+0
-4
lines changed

3 files changed

+0
-4
lines changed

src/refiners/training_utils/clock.py

-2
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,10 @@ def __init__(
2222
self,
2323
training_duration: TimeValue,
2424
gradient_accumulation: Step,
25-
lr_scheduler_interval: Iteration | Epoch,
2625
verbose: bool = True,
2726
) -> None:
2827
self.training_duration = training_duration
2928
self.gradient_accumulation = gradient_accumulation
30-
self.lr_scheduler_interval = lr_scheduler_interval
3129
self.verbose = verbose
3230
self.start_time = None
3331
self.end_time = None

src/refiners/training_utils/trainer.py

-1
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,6 @@ def clock(self, config: ClockConfig) -> TrainingClock:
139139
return TrainingClock(
140140
training_duration=self.config.training.duration,
141141
gradient_accumulation=self.config.training.gradient_accumulation,
142-
lr_scheduler_interval=self.config.lr_scheduler.update_interval,
143142
verbose=config.verbose,
144143
)
145144

tests/training_utils/test_trainer.py

-1
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,6 @@ def training_clock() -> TrainingClock:
205205
return TrainingClock(
206206
training_duration=Epoch(5),
207207
gradient_accumulation=Step(1),
208-
lr_scheduler_interval=Epoch(1),
209208
)
210209

211210

0 commit comments

Comments
 (0)