rizgiak's picture
initial commit
737c08b
raw
history blame contribute delete
151 Bytes
# Define callbacks here
from pytorch_lightning.callbacks import EarlyStopping
early_stopping = EarlyStopping(monitor="loss", min_delta=0, patience=3)