How to solve target tensor size and input tensor size mismatch

[xqc-qc]

The type of my own data set is 6 categories, and the pre-training model is 80 categories. I changed the config to 6 categories, and I still get an error. It seems that the target tensor size does not match the input tensor size. How can I correct this problem?