分类：callbacks

2020
10-08

浅谈keras.callbacks设置模型保存策略

如下所示：keras.callbacks.ModelCheckpoint(self.checkpoint_path,verbose=0,save_weights_only=True,mode="max",save_best_only=True),默认是每一次poch,但是这样硬盘空间很快就会被耗光.将save_best_only设置为True使其只保存最好的模型,值得一提的是其记录的acc是来自于一个monitor_op,其默认为"val_loss",其实现是取self.best为-np.Inf.所以,第一次的训练结果总是被保存.mode模式自动为auto和max一样,还有一个min的选项... 继续阅读 >