Hi,
It depends on what loss function the model has defined. As explained on Stackoverflow 1, your labels must either be 1-dimensional or 2-dimensional, depending on the loss function being used.
i.e. they must either be of shape (batch_size,) in which case they contain the class indices, or of shape (batch_size, num_labels), in which case they contain one-hot encoded targets.