• 周五. 12月 2nd, 2022

5G编程聚合网

5G时代下一个聚合的编程学习网

热门标签

模型权重的保存与加载 回调函数的使用

admin

11月 28, 2021
import tensorflow as tf
import numpy as np
#保存权重
model_file = "./model/cifarmodel.h5"
model.save_weights(model_file)
print("已保存模型权重!")

#加载权重
try:
    model.load_weights(model_file)
    print("权重加载成功!")
except:
    print("权重加载失败!")
    
    
    
#通过回调函数保存权重
check_points = "./model/cifar10.{epoch:02d} - {val_loss: .4f}.h5"
callbacks = [ tf.keras.callbacks.ModelCheckpoint(filepath = check_points, 
                                                 save_weights_only = True,
                                                 verbose = 0, save_freq = 'epoch'),
             tf.keras.callbacks.EarlyStopping(monitor = "val_loss", patience = 3),
             # 早停,防止过拟合,监控值为val_loss,如果连续三个周期的值都越来越差,则停止保存check_points
]
#回调函数的应用
model.fit(train_x, train_y, validation_split = 0.3, epoch = 5, batch_size = 100, callbacks = callbacks, verbose = 2)
    
自己选择的路,跪着也要走完。朋友们,虽然这个世界日益浮躁起来,只要能够为了当时纯粹的梦想和感动坚持努力下去,不管其它人怎么样,我们也能够保持自己的本色走下去。

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注