I am trying to run Video Vision Transformer (ViViT) code with my dataset but getting an error using CrossEntropyLoss from Pytorch as the Loss function. There are 6 classes I have: ['Run', 'Sit', 'W...