Hey @jackieliu930,

Yes, the labels need to be int values
Yes you need to modify the .from_pretrained method here: notebooks/train.py at 3fdb8bd61ed2f2b499dcd55034b1ee58be5cfabb · huggingface/notebooks · GitHub 3


You could also use the run_glue.py from examples 2 using the git_config then you don’t need to provide your own training script.