Solved by ptrblck in post #2 
              
              
                Depending on the type your __getitem__ method from the Dataset returns, the default_collate will use torch.stack, create a torch.tensor etc.