Skip to content

Commit 628437c

Browse files
authored
fix ReduceLROnPlateau import from lightning (#21)
1 parent 39ab0d7 commit 628437c

File tree

1 file changed

+1
-1
lines changed
  • src/finetuning_scheduler/strategy_adapters

1 file changed

+1
-1
lines changed

src/finetuning_scheduler/strategy_adapters/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222

2323
import torch
2424

25+
from torch.optim.lr_scheduler import ReduceLROnPlateau
2526
from lightning.fabric.utilities import rank_zero_info, rank_zero_warn
26-
from lightning.fabric.utilities.types import ReduceLROnPlateau
2727
from lightning.pytorch import LightningModule, Trainer
2828
from lightning.pytorch.callbacks import Callback
2929
from lightning.pytorch.callbacks import BaseFinetuning

0 commit comments

Comments
 (0)