But shouldnt it be just 1? Because i train with my 1-unit dataset? And also, what should nb_classes be at my CNN? Since when thinking about a real-training later, there are most likely “infinite” classes. (I know its not infinite but reaaaally big). I am sorry for all my missunderstanding, i am very new too pytorch.
also the cnn was already given and i am not sure how to change the output that it fits like this.