模型权重记录与恢复

由admin

11 月 28, 2021 code

import tensorflow

logdir = './logs'
checkpoint_path = './checkpoint/Titanic.{epoch:02d}-{val_loss:.2f}.ckpt' #路径为当前目录下的checkpoint子目录，后边为命名规则
callbacks = [tf.keras.callbacks.TensorBoard(log_dir = logdir, histogram_freq = 2), #参数一：日志文件目录， 参数二：直方图频率为2
             tf.keras.callbacks.ModelCheckpoint(filepath = checkpoint_path, save_weights_only = True, verbose = 1, period = 5)]
                                                #参数一：ckpt文件的路径，参数二：仅保存模型的权重，参数三：保存时的输出信息，参数四：周期

logdir = './logs'
checkpoint_path = './checkpoint/Titanic.{epoch:02d}-{val_loss:.2f}.ckpt'
checkpoint_dir = os.path.dirname(checkpoint_path) #去掉文件名，返回目录
latest = tf.train.latest_checkpoint(checkpoint_dir) #找到最新的checkpoint
model.load_weights(latest) #加载权重