如何使用TensorFlow的Keras API为每个时代的保存模型生成唯一的名称

Question

我正在使用fit_generator()训练模型，并希望为每个时期保存的权重生成唯一的名称

已尝试：：查看

后面的代码

代码：

model_path = '.\checkpoints\cp{}.ckpt'.format(time())
cp_callback = tf.keras.callbacks.ModelCheckpoint(model_path, 
                                                 verbose=1,
                                                  period=2)

model.fit_generator(..........,callbacks=[cp_callback])

期望：生成唯一的检查点名称
例如epoch_4.ckpt或epoch_5.ckpt
实际：每次保存时，都会覆盖现有的检查点

Answer 1

您可以尝试将epoch变量添加到filepath变量中

filepath = ".\checkpoints\cp-{epoch:02d}.hdf5"
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath, verbose=1, period=2)

Keras文档here已经提到了这一点。