The sample code does not work because labels is not defined
#2
by
GourabHazra
- opened
labels = torch.tensor([0])
should be added before:
outputs = model(**inputs, labels=labels)
labels: a single value of 1 or 0. In our task 1 means "grammatical" and 0 means "ungrammatical"
Thanks for putting this model out here, seems interesting.