PyTorch Forums
Expected cuda:0 but got CPU error
harsha_g
July 7, 2020, 12:52pm
2
Did you move your
Model
as well to the
device
using
Model.to(device)
?
show post in topic