那些年，Pytorh的坑（持续更新）

支持那些致力于可持续生产的企业。 #生活技巧# #环保生活方式# #可持续消费#

        本博客和大家分享一些本人在使用pytorch时候经历过的一些坑，将持续更新。

1. 报错：

RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'weight' 1

任务：需要实现从test_loader中随机获取一张图片，传入已经训练好的model中，运行以下代码报上述错误。

dataiter = iter(test_loader) images, labels = dataiter.next() images = Variable(images) print model(images) 1234 解决办法：加入这句话images = images.type(torch.cuda.FloatTensor)

dataiter = iter(test_loader) images, labels = dataiter.next() images = Variable(images) images = images.type(torch.cuda.FloatTensor) print model(images) 1234567

2.报错

urllib2.URLError: <urlopen error [Errno 1] _ssl.c:510: error:14077410:SSL routines:SSL23_GET_SERVER_HELLO:sslv3 alert handshake failure> 1

任务：从torchvision.models中下载已训练好的模型，会报如上错误，相关代码如下：

net = models.vgg16(pretrained=True) 1

本人在网上查了很久，有人说时python版本的原因，本人python版本为2.7.6，version需要大于2.7.9.但是重装更新python太麻烦了。于是采用以下方法进行解决：replace('https://', 'http://')

from torchvision.models.vgg import model_urls import torchvision model_urls['vgg16'] = model_urls['vgg16'].replace('https://', 'http://') vgg16 = torchvision.models.vgg16(pretrained=True) 12345

3. 心得

使用pytorch加载CIFAR10数据集的详情

train_dataset = torchvision.datasets.CIFAR10( root=DATA_PATH, train=True, transform=transform_train, download=True ) test_dataset = torchvision.datasets.CIFAR10(root='../../data/', train=False, transform=transform) 1234

classes = ('plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck') 1

cifar10数据集经常做10分类，训练集有50000张图片，测试集有10000张。整个数据集有60000张图片。

网址：那些年，Pytorh的坑（持续更新） https://www.yuejiaxmz.com/news/view/500913

相关内容

Shopify收款那些坑（三）——派安盈与2checkout
盘点那些“比较坑”的汽车装饰配件
2018新的一年，家里的哪些老物件还会持续升值？
轻松一招教你避开网上购物那些大坑！
儿童减肥的那些坑，你入了几个？
那些修复书画的年轻人
那些想“避雷”二手奢侈品假货的年轻人，先被鉴定“坑”了？
可持续不持有 绿色低碳改造法让家更年轻更环保
更加可持续的十种生活方式
可持续时尚｜新兴材料赋能可持续鞋履创新

随便看看