python - tf.keras.callbacks.ModelCheckpoint 与 tf.train.Checkpoint

Question

我对 TensorFlow 世界有点陌生，但已经用 Keras 编写了一些程序。由于 TensorFlow 2 正式类似于 Keras，我很困惑 tf.keras.callbacks.ModelCheckpoint 和 tf.train.Checkpoint 之间的区别。如果有人能阐明这一点，我将不胜感激。

Answer 1

这取决于是否需要自定义训练循环。在大多数情况下，它不是，您只需调用 model.fit() 并传递 tf.keras.callbacks.ModelCheckpoint。如果您确实需要编写自定义训练循环，则必须使用 tf.train.Checkpoint(和 tf.train.CheckpointManager)，因为没有回调机制。