new to Lightning so this question might sound a little naive (sorry about that).
I have a training script performing a straight image classification task.
I’m using a ReduceLROnPlateau scheduler and I would like to stop the training if the learning rate reaches a minimum value.
To do that, I tried inserting an
EarlyStopping(monitor='lr', mode='min', stopping_threshold=min_lr_value) callback, logging the learning rate with a
However, the training fails after the first epoch with the following error:
RuntimeError: Early stopping conditioned on metric `lr` which is not available. Pass in or modify your `EarlyStopping` callback to use any of the following: `train_loss`, `val_loss`
The learning rate gets correctly logged on Wandb though with the
Am I missing something? Or do I have to explicitly log each metric of interest for the
EarlyStopping callback to be able to monitor it?
Thanks for any help you can give me!