Hi, All.
This is my simple GPU code in PyTorch.
How can I change this in CPU code?
filename=r’/Users/dti/Desktop/PyTorch4testAccuracy/hymenoptera_data/test/bees/1.jpg’
img = skimage.io.imread(filename)
x = V(centre_crop(img).unsqueeze(0), volatile=True).cuda()
model = models.dict’resnet18’
model = torch.nn.DataParallel(model).cuda()
model = torch.load(‘model5.pth’)
logit = model(x)
print(logit)
Thanks in advance .