如何从tensorflow检查点文件中正确恢复网络训练？

import tensorflow as tf

# 定义网络结构
# ...

model = YourModel()

# 定义优化器和损失函数
# ...

checkpoint = tf.train.Checkpoint(model=model, optimizer=optimizer)
manager = tf.train.CheckpointManager(checkpoint, checkpoint_directory, max_to_keep=3)

# 恢复网络训练
latest_checkpoint = manager.latest_checkpoint
if latest_checkpoint:
    checkpoint.restore(latest_checkpoint)
    print("从检查点文件 {} 中成功恢复网络训练。".format(latest_checkpoint))
else:
    print("未找到检查点文件，无法恢复网络训练。")