pytorch dataloader default_collate argument use with to(device)

Question

Ive been trying to integrate the to(device) inside my dataloader using to(device) as seen in https://github.com/pytorch/pytorch/issues/11372

I defined it on FashionMNIST