Problems with number of classes, adjust CrossEntropyLoss?

Actually, it depends on you. :wink: