3578019be301ae9d8e9b7518daa5fb113d00d2db,open_seq2seq/utils/hooks.py,RunEvaluationHook,after_run,#RunEvaluationHook#Any#Any#,147

Before Change

    inputs_per_batch, outputs_per_batch = [], []
    total_loss = 0.0

    for cnt, feed_dict in enumerate(
      self._model.data_layer.iterate_one_epoch(cross_over=True)
    ):
      loss, inputs, outputs = run_context.session.run(
        self._fetches, feed_dict,
      )
      inputs_per_batch.append(inputs)
      outputs_per_batch.append(outputs)
      total_loss += loss

    total_loss /= (cnt + 1)
    deco_print("Validation loss: {:.4f}".format(total_loss), offset=4)
    dict_to_log = self._model.maybe_evaluate(
      inputs_per_batch,

After Change

      return
    self._timer.update_last_triggered_step(self._iter_count - 1)

    if not self._model.on_horovod or self._model.hvd.rank() == 0:
      deco_print("Running evaluation on a validation set:")

    inputs_per_batch, outputs_per_batch, total_loss = get_batches_for_epoch(
      self._model, run_context.session, compute_loss=True,
    )

Instances