Wie kann ich Keras mitteilen, dass das Training basierend auf dem Verlustwert beendet wird?

Zur Zeit benutze ich den folgenden Code:

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)

Es weist Keras an, das Training abzubrechen, wenn sich der Verlust 2 Epochen lang nicht verbessert hat. Aber ich möchte aufhören zu trainieren, nachdem der Verlust kleiner als ein konstanter "THR" wurde:

if val_loss < THR:
    break

Ich habe in der Dokumentation gesehen, dass es die Möglichkeit gibt, einen eigenen Rückruf durchzuführen:http: //keras.io/callbacks Aber nichts gefunden, wie man den Trainingsprozess stoppen kann. Ich brauche einen Rat