Multi-label Image Classification Colab Notebook Error

[TonyAssi]

This notebook doesn't work https://colab.research.google.com/github/johko/computer-vision-course/blob/main/notebooks/Unit%203%20-%20Vision%20Transformers/fine-tuning-multilabel-image-classification.ipynb

When running this line of code: notebook_launcher(train, (model_name,8,5,5e-5), num_processes = 2)

The following error: File "", line 10, in train_transforms labels = torch.tensor(batch['classes']) ValueError: expected sequence of length 1 at dim 1 (got 2)