from keras.callbacks import ModelCheckpoint
'''
如果验证损失下降, 那么在每个训练轮之后保存模型。
'''
checkpointer = ModelCheckpoint(filepath='weights.hdf5', verbose=1, save_best_only=True)
model.fit_generator(train_generator, epochs=50,validation_data=test_generator,callbacks=[checkpointer])
keras保存断点模型与加载断点
最新推荐文章于 2022-03-22 22:47:03 发布