Yes, I believe that got me confused. So after the model is trained, let’s say I’d like to test it on an image using pred = model(img). How can I run this prediction on GPU, given that img is on GPU as well?
Yes, I believe that got me confused. So after the model is trained, let’s say I’d like to test it on an image using pred = model(img). How can I run this prediction on GPU, given that img is on GPU as well?