Huggingface BERT SequenceClassification - ValueError: too many values to unpack (expected 2)

Question

The class labels for the two class model is 0, 1, 0, 0, etc. There is only one label per input sequence. The labels are set in a python list and converted to torch.Tensor. (read