Skip to content

Commit

Permalink
fix typing for scheduler
Browse files Browse the repository at this point in the history
  • Loading branch information
helpmefindaname committed Oct 11, 2023
1 parent 5631d27 commit df71f85
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions flair/trainers/plugins/functional/anneal_on_plateau.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ def __init__(
self.anneal_factor = anneal_factor
self.patience = patience
self.initial_extra_patience = initial_extra_patience
self.scheduler: AnnealOnPlateau

def store_learning_rate(self):
optimizer = self.trainer.optimizer
Expand Down

0 comments on commit df71f85

Please sign in to comment.