python - 我可以在使用 slim.learning.train 时获取张量吗？

Question

感谢TF-Slim，构建新模型更容易

但是在训练模型时，slim.learning.train 似乎无法获取我需要检查模型的张量。

它只能返回损失（训练操作），下面的代码在 tensorflow/contrib/slim/python/slim/learning.py 中，它显示了损失是如何打印的。

if sess.run(train_step_kwargs['should_log']):
  logging.info('global step %d: loss = %.4f (%.3f sec/step)',
               np_global_step, total_loss, time_elapsed)

有什么方法可以获取张量或只打印它的值吗？

Answer 1

假设您有 tensor t，那么您可以在每次执行涉及使用的图形时打印它，方法t是将张量与tf.Print()打印函数相关联，如下所示：

t = tf.Print(t, [t])