hi ,I am encounted this problem too,because some classes of my dataset just have one ,so it either in trainset or testset.The label is just like this:trainset_label[0,2,3,5],testset_labe[1,2,4],so I set the num_classes=6,how can I fix this problem?thank you very much!