We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 39ab0d7 commit 628437cCopy full SHA for 628437c
src/finetuning_scheduler/strategy_adapters/base.py
@@ -22,8 +22,8 @@
22
23
import torch
24
25
+from torch.optim.lr_scheduler import ReduceLROnPlateau
26
from lightning.fabric.utilities import rank_zero_info, rank_zero_warn
-from lightning.fabric.utilities.types import ReduceLROnPlateau
27
from lightning.pytorch import LightningModule, Trainer
28
from lightning.pytorch.callbacks import Callback
29
from lightning.pytorch.callbacks import BaseFinetuning
0 commit comments