yolov3代码涉及到的Keras.callbacks类

2019-08-17  本文已影响0人  dc_3

官网解释

Keras.callbacks

回调函数是一组在训练的特定阶段被调用的函数集,你可以使用回调函数来观察训练过程中网络内部的状态和统计信息。通过传递回调函数列表到模型的.fit()中,即可在给定的训练阶段调用该函数集中的函数。

ModelCheckpoint:

该回调函数将在每个epoch后保存模型到filepath

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', 
verbose=0, save_best_only=False, 
save_weights_only=False, mode='auto', period=1)

ReduceLROnPlateau:

当学习停滞时,减少2倍或10倍的学习率常常能获得较好的效果。该回调函数检测指标的情况,如果在patience个epoch中看不到模型性能提升,则减少学习率。学习率对模型训练的影响解释

keras.callbacks.ReduceLROnPlateau(monitor='val_loss', 
factor=0.1, patience=10, verbose=0, mode='auto', 
epsilon=0.0001, cooldown=0, min_lr=0)

EarlyStopping:

当监测值monitor不再改善时,该回调函数将中止训练

keras.callbacks.EarlyStopping(monitor='val_loss', patience=0, 
verbose=0, mode='auto')

yolov3代码->train.py涉及到的:

ModelCheckpoint:

checkpoint = ModelCheckpoint(log_dir + 'ep{epoch:03d}-

loss{loss:.3f}-val_loss{val_loss:.3f}.h5',

        monitor='val_loss', save_weights_only=True, 

save_best_only=True, period=3)

参数解释:

ReduceLROnPlateau:

    reduce_lr = ReduceLROnPlateau(monitor='val_loss', 

factor=0.1, patience=3, verbose=1)

参数解释:

EarlyStopping:

early_stopping = EarlyStopping(monitor='val_loss', 

min_delta=0, patience=10, verbose=1)

参数解释:

上一篇 下一篇

猜你喜欢

热点阅读