Solved by ptrblck in post #4 
              
              
                Could you check the type of outputs and labels? 
I guess labels is passed as a tuple instead of a tensor. 
If so, you could return labels in your __getitem__ as: 
torch.tensor(self.labels[idx])