Changing labels in Dataloader

Hi all I have used the following code to select a piece of data. Everything goes well before the data is passed to the Dataloader. For example I want to select the data from ‘cat’, ‘dog’ and ‘deer’ classes with target labels 3, 4 and 5. The target data had selected successfully before passing to the Dataloader after passing the labels changed to the 0, 1 and 2. Is there any problem here?


Your description of selecting a subset of classes and remap them to [0, nb_classes-1] sounds valid. Do you see any issues running your code? If so, could you post the error messages and stacktraces, please?

Hi @ptrblck thanks for your reply. There is not any error or warring.