Hello everyone,
So, I have a problem and I hope that you guys can help me figure out a solution for this.
so I wrote a model and trained it and I am trying to test it out.
and I get this error so here is the full sityuation:
device = 'cuda' if torch.cuda.is_available() else 'cpu'
print(device)
Model.eval()
    with torch.no_grad():
        total_correct = 0
        size_of_set = 0
    for images, labels in DataLoader:
        images = images.to(device)
        print(images.device)
        labels = labels.to(device)
        preds = Model(images)
        total_correct+= get_correct(preds,labels)

So actually, the first print shows that cuda is available.
but the second one shows that the images are given to the cpu…
This is weird and then I get this error:
RuntimeError: expected device cuda:0 but got device cpu