Hi, I am using the same model with two following data loading strategy:




train_loader = DataLoader(TensorDataset(train_0, train_t), batch_size=batch_size, 
                        shuffle=True, drop_last=False)





train_loader = DataLoader(torch.cat((train_0, train_t), dim=1), batch_size=batch_size, 
                        shuffle=True, drop_last=False)

I trained both cases for 2 epochs, and case 1 took 30 seconds while case 2 took only 13 s.  Does anyone know why there would be a significant difference between these two?
Thanks!