Question

现在我正尝试在各种情况下使用tf.train.Checkpoint而不是tf.train.Saver。

当我尝试在MonitoredTrainingSession中使用Checkpoint时，我注意到 Checkpoint.save更改tf.Graph。因此，我不能如下使用Checkpoint。

a = tf.Variable([3], name="a")
checkpoint = tf.train.Checkpoint(model=a)

with tf.train.MonitoredTrainingSession() as sess:
    checkpoint.save(file_prefix="parameter/ckpt", session=sess)

我收到一个错误“ RuntimeError：图形已完成，无法修改”。

我可以将Checkpoint与MonitoredTrainingSession一起使用吗？

Answer 1

我发现了一种怪异的技术，该技术通过临时会话调用checkpoint.write来初步构建图形。

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    checkpoint.save("./parameter/sample")

with tf.train.MonitoredTrainingSession() as sess:
    checkpoint.save(file_prefix="parameter/ckpt", session=sess)

有什么办法可以将tf.train.Checkpoint与MonitoredTrainingSession一起使用吗？

1 个答案: