Skip to content

Instantly share code, notes, and snippets.

@doleron

doleron/medium_early_stopping.py

Last active April 16, 2023 13:06
Show Gist options
  • Save doleron/0bb9b95b0930929d6143df9ae990ae43 to your computer and use it in GitHub Desktop.
Save doleron/0bb9b95b0930929d6143df9ae990ae43 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
medium_early_stopping.py
early_stop = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience = 10)
history = model.fit(train_ds, validation_data=validation_ds, epochs=EPOCHS, callbacks=[early_stop])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment