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)
This image shows how the target is read from the dataframe and the shape of the target is printed in terminal.