292ec4cba5d5f6f8beca446fb18b2797ac5ae9d2,official/vision/beta/tasks/video_classification.py,VideoClassificationTask,inference_step,#VideoClassificationTask#,201

Before Change

  def inference_step(self, inputs, model):
    Performs the forward step.
    return model(inputs, training=False)

After Change

    Performs the forward step.
    outputs = model(inputs, training=False)
    if self.task_config.train_data.is_multilabel:
      outputs = tf.math.sigmoid(outputs)
    else:
      outputs = tf.math.softmax(outputs)
    return outputs

Instances