Atualmente, eu uso o seguinte 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)

Ele diz a Keras para parar o treinamento quando a perda não melhora há 2 épocas. Mas quero parar de treinar após a perda se tornar menor do que um constante "THR":

if val_loss < THR:
    break

Vi na documentação que há possibilidade de fazer seu próprio retorno de chamada:http://keras.io/callbacks/ Mas nada encontrou como interromper o processo de treinamento. Eu preciso de um conselho.