有没有办法抑制tf.train.MonitoredTrainingSession()中的特定钩子?

时间:2018-03-23 11:10:30

标签: python tensorflow hook

我想在tf.train.MonitoredTrainingSession中压缩一个钩子,同时运行其他钩子。有没有办法做到这一点?目前,当钩子在sess.run()上运行一个不应该运行的操作时,我遇到了错误。我正在寻找的运营商的一个例子是:

with tf.train.MonitoredTrainingSession(
   hooks=[hook1, hook2, hook3, hook4]
) as sess:
   sess.run_partial_hook(fetches=[...], suppress=[hook2, hook3], feed_dict={...})

要添加一些信息,我知道有一个sess.run_step_fn运行会话而没有添加钩子。我还考虑过手动编写钩子的可能性,以便在需要时跳过sess.run()

0 个答案:

没有答案