check your num_classes? what you have put code for this.
check this post, same error output, see if it helps understand
Otherwise share your this part of the code in personal DM. Click on my name and then message. Do not post your code here.
Regards
DP