誰かがcifar10_train.pyの列車機能をテンソルフローのcifar10チュートリアルから説明できますか

Question

私はhttps://github.com/tensorflow/models/tree/master/tutorials/image/cifar10からcifar10チュートリアルに従っています。 このプロジェクトには6つのクラスがあります。インターネットを検索した後、私はcifar10.pyとcifar10_input.pyクラスを理解しました。しかし、私はcifar10_train.pyで列車の機能を理解できません。 cifar10_train.pyクラスの列車機能は次のとおりです。

def train(): 
with tf.Graph().as_default(): 
    global_step = tf.contrib.framework.get_or_create_global_step() 

    # get images and labels for cifar 10 
    # Force input pipeline to CPU:0 to avoid operations sometime ending on 
    # GPU and resulting in a slow down 

    with tf.device('/cpu:0'): 
     images, labels = cifar10.distorted_inputs() 

    logits = cifar10.inference(images) 

    loss = cifar10.loss(logits, labels) 

    train_op = cifar10.train(loss, global_step) 

    class _LoggerHook(tf.train.SessionRunHook): 

     def begin(self): 
      self._step = -1 
      self._start_time = time.time() 

     def before_run(self, run_context): 
      self._step += 1 
      return tf.train.SessionRunArgs(loss) 

     def after_run(self, run_context, run_values): 
      if self._step % FLAGS.log_frequency == 0: 
       current_time = time.time() 
       duration = current_time - self._start_time 
       self._start_time = current_time 

       loss_value = run_values.results 
       examples_per_sec = FLAGS.log_frequency * FLAGS.batch_size/duration 
       sec_per_batch = float(duration/FLAGS.log_frequency) 

       format_str = ('%s: step %d, loss = %.2f (%.1f examples/sec; %.3f ' 
           'sec/batch)') 
       print(format_str % (datetime.now(), self._step, loss_value, 
            examples_per_sec, sec_per_batch)) 

    with tf.train.MonitoredTrainingSession(
      checkpoint_dir=FLAGS.train_dir, 
      hooks=[tf.train.StopAtStepHook(last_step=FLAGS.max_steps), 
        tf.train.NanTensorHook(loss), 
        _LoggerHook()], 
      config=tf.ConfigProto(
       log_device_placement=FLAGS.log_device_placement)) as mon_sess: 
     while not mon_sess.should_stop(): 
      mon_sess.run(train_op) 

誰かが_LoggerHookクラスで何が起こっているのか説明できますか？

Answer 1

トレーニング時にロスを記録するには、MonitoredSessionとSessionRunHookを使用します。

_LoggerHookは、以下の順序で実行SessionRunHookの実装です：

call hooks.begin() 
    sess = tf.Session() 
    call hooks.after_create_session() 
    while not stop is requested: 
    call hooks.before_run() 
    try: 
     results = sess.run(merged_fetches, feed_dict=merged_feeds) 
    except (errors.OutOfRangeError, StopIteration): 
     break 
    call hooks.after_run() 
    call hooks.end() 
    sess.close() 

それはhereからです。

session.runの前にlossのデータを収集し、定義済みの形式のlossを出力します。

チュートリアル：https://www.tensorflow.org/tutorials/layers

が、これは期待している願っています。