¿Cómo decirle a Keras que deje de entrenar en función del valor de la pérdida?

Actualmente uso el siguiente código:

callbacks = [
    EarlyStopping(monitor='val_loss', patience=2, verbose=0),
    ModelCheckpoint(kfold_weights_path, monitor='val_loss', save_best_only=True, verbose=0),
]
model.fit(X_train.astype('float32'), Y_train, batch_size=batch_size, nb_epoch=nb_epoch,
      shuffle=True, verbose=1, validation_data=(X_valid, Y_valid),
      callbacks=callbacks)

Le dice a Keras que deje de entrenar cuando la pérdida no mejoró durante 2 épocas. Pero quiero dejar de entrenar después de que la pérdida se hizo más pequeña que un "THR" constante:

if val_loss < THR:
    break

He visto en la documentación que existe la posibilidad de hacer su propia devolución de llamada:http://keras.io/callbacks/ Pero nada encontró cómo detener el proceso de entrenamiento. Necesito un consejo.