有什么方法可以使用吗列车检查站有监督训练课吗?

2024-04-27 03:23:24 发布

您现在位置:Python中文网/ 问答频道 /正文

现在我想用列车检查站而不是tf.列车保护器在各种情况下。你知道吗

当我尝试将Checkpoint与MonitoredTrainingSession一起使用时,我注意到 那个检查点.保存变化tf.图形. 因此,我不能像下面那样使用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一起使用吗?你知道吗


Tags: name图形modeltf情况trainvariable检查点
1条回答
网友
1楼 · 发布于 2024-04-27 03:23:24

我发现了一个奇怪的方法检查点写入用一个临时会话来初步构造一个图。你知道吗

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)

相关问题 更多 >