Skip to content

Instantly share code, notes, and snippets.

@VedPDubey

VedPDubey/callbacks.py

Created July 2, 2021 16:49
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save VedPDubey/4d906a9472dc91015633b57976840a5b to your computer and use it in GitHub Desktop.
Save VedPDubey/4d906a9472dc91015633b57976840a5b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
callbacks.py
checkpoint = ModelCheckpoint(
'./base.model',
monitor='val_accuracy',
verbose=1,
save_best_only=True,
mode='max',
save_weights_only=False,
save_frequency=1
)
earlystop = EarlyStopping(
monitor='val_loss',
min_delta=0.001,
patience=30,
verbose=1,
mode='auto'
)
opt1 = tf.keras.optimizers.Adam()
callbacks = [checkpoint,earlystop]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment