Solved by Joshua_Pan in post #3 
              
              
                I think it’s because of the difference between the output of   __getitem__ method of Concated two datasets( train_set and pseudo_set).  __getitem__  of train_set yields (image, int label) while __getitem__ of pseudo_set yields (image, tensor label). After two datasets are concated, a set of indices …