File size: 151 Bytes
4c41a36
 
 
 
1
2
3
4
5
# Define callbacks here
from pytorch_lightning.callbacks import EarlyStopping

early_stopping = EarlyStopping(monitor="loss", min_delta=0, patience=3)