随笔分类 -  Pytorch

摘要:安装或者更新完pytorch后,运行不了,显示错误: 错误显示为:libmkl_intel_lp64.so: cannot open shared object file: No such file or directory libmkl_intel_lp64.so是一个动态链接库文件,系统找不到。 阅读全文
posted @ 2019-05-11 14:04 denny402 阅读(15403) 评论(1) 推荐(2) 编辑
摘要:利用pytorch来构建网络模型有很多种方法,以下简单列出其中的四种。 假设构建一个网络模型如下: 卷积层--》Relu层--》池化层--》全连接层--》Relu层--》全连接层 首先导入几种方法用到的包: 第一种方法 这种方法比较常用,早期的教程通常就是使用这种方法。 第二种方法 这种方法利用to 阅读全文
posted @ 2017-09-25 18:40 denny402 阅读(35876) 评论(2) 推荐(4) 编辑
摘要:大部分的pytorch入门教程,都是使用torchvision里面的数据进行训练和测试。如果我们是自己的图片数据,又该怎么做呢? 一、我的数据 我在学习的时候,使用的是fashion-mnist。这个数据比较小,我的电脑没有GPU,还能吃得消。关于fashion-mnist数据,可以百度,也可以 点 阅读全文
posted @ 2017-09-14 12:54 denny402 阅读(85303) 评论(10) 推荐(11) 编辑
摘要:图片数据一般有两种情况: 1、所有图片放在一个文件夹内,另外有一个txt文件显示标签。 2、不同类别的图片放在不同的文件夹内,文件夹就是图片的类别。 针对这两种不同的情况,数据集的准备也不相同,第一种情况可以自定义一个Dataset,第二种情况直接调用torchvision.datasets.Ima 阅读全文
posted @ 2017-09-12 22:17 denny402 阅读(32777) 评论(7) 推荐(2) 编辑
摘要:#!/usr/bin/env python # -*- coding: utf-8 -*- __author__ = 'denny' __time__ = '2017-9-9 9:03' import torch import torchvision from torch.autograd import Variable import torch.utils.data.dataloader a... 阅读全文
posted @ 2017-09-11 19:26 denny402 阅读(17105) 评论(1) 推荐(1) 编辑