Solved by Mdhvince in post #3 
              
              
                Convert your data to float tensors should solve it: 
i.e 
dtype = torch.cuda.FloatTensor if torch.cuda.is_available() else torch.FloatTensor
for i, data in enumerate(dataloader):
    data['image'] = data['image'].type(dtype)
    data['label'] = data['label'].type(dtype) #Comment this line if it is a…