【发布时间】:2017-12-16 07:24:30
【问题描述】:
感谢 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)
有没有一些方法可以获取张量或只打印它的值?
【问题讨论】:
标签: python tensorflow tf-slim