reshape targets in eval

tensor2tensor/utils/t2t_model.py

@@ -1503,11 +1503,14 @@ def estimator_spec_eval(self, features, logits, labels, loss, losses_dict):
  # like actions or rewards.
  tf.logging.warning("No key %s in logits for evaluation." % k)
  else:
- # FATHOM
- # NOTE: right now the tf.logging.warning will not trigger.
- # ...consider if we should add
- eval_metrics[metric_name] = metric_fn(logits, features,
- features["targets"])
+ if hasattr(problem, 'max_docs_per_pack'):
+ features['targets'] = tf.reshape(features['targets'], [problem.max_docs_per_pack, -1])
+ else:
+ # FATHOM
+ # NOTE: right now the tf.logging.warning will not trigger.
+ # ...consider if we should add
+ eval_metrics[metric_name] = metric_fn(logits, features,
+ features["targets"])
  if isinstance(logits, dict):
  predictions = logits
  else: