pytorch搭建网络测试时报错Expected object of type torch.DoubleTensor but found type torch.FloatTensor...

用pytorch搭建网络测试时，代码报错如下：
Expected object of type torch.DoubleTensor but found type torch.FloatTensor for argument #2 ‘weight’

搭建的网络为AlexNet，测试代码如下

from torchvision import models

model = models.alexnet(pretrained=True)
x = np.random.rand(1,3,224,</