2017-07-12 2 views
0

おかげで、新しいモデルにslim.learning.trainを使用しているときにテンソルを取得できますか? TF-スリム用

を構築することが容易である。しかしモデルのトレーニングについて来るとき、slim.learning.trainは私が私のモデルをチェックする必要がありテンソルを取得することができないようです。

損失を返すことができます(トレーニングop)。以下のコードは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) 

テンソルをフェッチしたり、値を印刷する方法はありますか?

答えて

1

あなたはテンソルtを持っていると仮定すると、あなたはそれを次のようにtf.Print()印刷機能とテンソルを関連付けることによってtを使用することを含むグラフが実行されるたびに印刷することができます。

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

関連する問題