3578019be301ae9d8e9b7518daa5fb113d00d2db,open_seq2seq/utils/funcs.py,,evaluate,#Any#Any#,149

Before Change

  inputs_per_batch, outputs_per_batch = get_batches_for_epoch(model,
                                                              checkpoint)
  eval_dict = model.maybe_evaluate(inputs_per_batch, outputs_per_batch)
  deco_print("Finished evaluation")
  return eval_dict

After Change

def evaluate(model, checkpoint):
  inputs_per_batch, outputs_per_batch = restore_and_get_batches(model,
                                                                checkpoint)
  if not model.on_horovod or model.hvd.rank() == 0:
    eval_dict = model.maybe_evaluate(inputs_per_batch, outputs_per_batch)
    deco_print("Finished evaluation")
    return eval_dict
  else:
    return None

Instances