bf6c76a1a0c6cb5530f8a913fdb955726027597f,tensorflow_ranking/python/head.py,_RankingHead,create_estimator_spec,#_RankingHead#Any#Any#Any#Any#Any#,202

Before Change

        train_op = self._train_op_fn(regularized_training_loss)
      else:
        raise ValueError("train_op_fn and optimizer cannot both be None.")
      return tf.estimator.EstimatorSpec(
          mode=mode,
          predictions=logits,
          loss=regularized_training_loss,
          train_op=train_op)

After Change

      // Train.
      if mode == tf.estimator.ModeKeys.TRAIN:
        return tf.estimator.EstimatorSpec(
            mode=mode,
            loss=regularized_training_loss,
            train_op=_get_train_op(regularized_training_loss, self._train_op_fn,
                                   self._optimizer),
            predictions=logits)
      raise ValueError("mode={} unrecognized".format(mode))

Instances