Keras-基于用户输入的早期浇灌

我想知道是否有一种简单的方法可以根据用户输入而不是任何特定指标的监控来创建一种在喀拉斯早期停止的触发方法。

例如,我想发送一个键盘信号到执行培训的进程,以便它离开 fit_generator 函数并执行其余代码。

有什么想法吗?

编辑:根据@ankurgoel的回答,我编写了以下代码:

# Monitors the SIGINT (ctrl + C) to safely stop training when it is sent
flag = False
class TerminateOnFlag(Callback):
    """Callback that terminates training when the flag is raised.
    """
    def on_batch_end(self, batch, logs=None):
        if flag:    
            self.model.stop_training = True

def handler(signum, frame):
    logging.info('SIGINT signal received. Training will finish after this epoch')
    global flag
    flag = True

signal.signal(signal.SIGINT, handler) # We assign a specific handler for the SIGINT signal
terminateOnFlag = TerminateOnFlag()
callbacks.append(terminateOnFlag)

在哪里? callbacks 是我输入的回调列表 菲特发生器 .

在训练期间,当我发送 SIGINT 信号真的我明白了 SIGINT signal received. Training will finish after this epoch 但是,当时代结束时,什么也不会发生。怎么回事?