Method build_inputs

official/modeling/multitask/test_utils.py:122–123

(self, params)

    return [tf_keras.metrics.Accuracy(name="bar_acc")]

  def build_inputs(self, params):
    return mock_data("bar")

  def build_losses(self, labels, model_outputs, aux_losses=None) -> tf.Tensor:
    loss = tf_keras.losses.mean_squared_error(labels, model_outputs)