I’m trying to train a classification model: multi-class classification
The loss function used is nn.CrossEntropyLoss()
The batch size is 8 and the number of classes to classify is 6.
The target labels aren’t one-hot encoded - (Checked it)

Screenshot from 2021-11-02 09-43-591920×1080 361 KB

This image shows how the target is read from the dataframe and the shape of the target is printed in terminal.