keras训练时保持网络的方法

ps:keras如何保存每一个epoch的权重?写个博客记录一下,顺便给有相同困扰的朋友一个参考。

直接上示例:

from keras.callbacks import ModelCheckpoint
filepath = "H:/model/weights_{epoch:03d}-{val_acc:.4f}.h5"
checkpoint = ModelCheckpoint(
    filepath=filepath,
    monitor='val_accuracy',
    save_best_only=False,
    verbose=1,
    save_weights_only=False,
    period=1
)
history = lstmNet.fit(x_train,y_train, epochs=15,validation_data=(x_test,y_test), validation_freq=1,callbacks=[checkpoint])

主要是这三步,其中filepath是保存路径,这是官网给出的方法,save_best_only和save_weights_only设置为False,否则只保存最佳模型或者只保存参数,最后一定要记得在fit函数中调用callbacks。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值