model.to(‘cuba’) should be changed to vgg16.to('cuda')
model.to(‘cuba’) should be changed to vgg16.to('cuda')