如何从已加载的检查点重置优化器状态

import tensorflow as tf

# 创建一个模型
model = tf.keras.Sequential([
    tf.keras.layers.Dense(64, activation='relu'),
    tf.keras.layers.Dense(10, activation='softmax')
])

# 创建一个优化器
optimizer = tf.keras.optimizers.Adam()

# 编译模型
model.compile(optimizer=optimizer, loss='sparse_categorical_crossentropy', metrics=['accuracy'])

# 保存检查点
checkpoint_path = "checkpoint.ckpt"
model.save_weights(checkpoint_path)

# 加载已保存的检查点
model.load_weights(checkpoint_path)

# 创建一个新的优化器对象，并设置为模型的优化器
new_optimizer = tf.keras.optimizers.Adam()
model.compile(optimizer=new_optimizer, loss='sparse_categorical_crossentropy', metrics=['accuracy'])