How to get the best model when using EarlyStopping callback in Keras?

后端 未结 2 1594
栀梦
栀梦 2020-12-10 21:12

I am training a neural network with Keras using EarlyStopping based on val_acc and patience=0. EarlyStopping stops the tr

2条回答
  •  独厮守ぢ
    2020-12-10 21:30

    In Keras 2.2.3, a new argument called restore_best_weights have been introduced for EarlyStopping callback that if set to True (defaults to False), it would restore the weights from the epoch with the best monitored quantity:

    restore_best_weights: whether to restore model weights from the epoch with the best value of the monitored quantity. If False, the model weights obtained at the last step of training are used.

提交回复
热议问题