And also what of the case if a third dataset had only classes “dog” and “pig” and I concatenated it with the first and second? Does ConcatDataset() map that pig should have class index 2 instead of 1?