Train end-to-end text classication on sagemaker

Amazon SageMaker
2

Hey @jackieliu930,

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

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