Solved by albanD in post #2 
              
              
                Hi, 
After a quick look at the sst dataset in torchtext, it seems that the number of labels for SST is not the same as IMDB. You need to resize your output layer accordingly.