![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch学习
黄鑫huangxin
深度学习,多媒体分析与理解
知乎专栏:https://www.zhihu.com/people/mu-xi-jin-39/columns
Github: https://github.com/BonnieHuangxin
展开
-
python之matplotlib绘制散点
使用scatter()绘制散点图#随意给出一些点x_train = np.array([[3.3],[4.4],[5.5],[6.71],[6.93],[4.168],[9.776],[6.182],[7.59],[2.167]],dtype=np.float32)y_train = np.array([[1.7],[2.76],[2.09],[3.19],[1.694],[1.573]...原创 2018-08-18 17:23:20 · 512 阅读 · 0 评论 -
pytorch中 max()、view()、 squeeze()、 unsqueeze()
总结:max():max(a),用于一维数据,求出最大值。 max(a,0),计算出数据中一列的最大值,并输出最大值所在的行号。 max(a,1),计算出数据中一行的最大值,并输出最大值所在的列号。a.view(i,j)表示将原矩阵转化为i行j列的形式 a.view(i,-1)表示不限制列数,输出i行 a.view(-1,j)表示不限制行数,输出j列a.squeeze(i)...原创 2018-08-22 10:51:30 · 494 阅读 · 0 评论 -
Pytorch之提取模型中的某一层
modules()会返回模型中所有模块的迭代器,它能够访问到最内层,比如self.layer1.conv1这个模块,还有一个与它们相对应的是name_children()属性以及named_modules(),这两个不仅会返回模块的迭代器,还会返回网络层的名字。方法如下:new_model = nn.Sequential(*list(model.children())[:2] 取模型...原创 2018-08-22 22:13:07 · 19461 阅读 · 4 评论 -
pytorch之权重初始化
因为权重是一个Variable,所以只需要取出其中的data属性,然后对它进行所需要的处理。for m in model.modules(): if isinstance(m,nn.Conv2d): nn.init.normal(m.weight.data) nn.init.xavier_normal(m.weight.data) nn...原创 2018-08-23 12:03:48 · 3252 阅读 · 0 评论 -
Python 图像库PIL
1、简单实用Image函数从文件加载图像,用Image函数的open方法>>> from PIL import Image>>> im = Image.open("hopper.ppm")如果成功,这个函数将返回一个图像对象。现在您可以使用实例属性来检查文件内容>>> from __future__ import print_fun...转载 2018-08-28 16:33:21 · 242 阅读 · 0 评论 -
基于Pytorch的MLP模块实现
MLP分类效果一般好于线性分类器,即将特征输入MLP中再经过softmax来进行分类。具体实现为将原先线性分类模块:self.classifier = nn.Linear(config.hidden_size, num_labels)替换为:self.classifier = MLP(config.hidden_size, num_labels)并且添加MLP模块: ...原创 2019-03-04 12:06:06 · 13969 阅读 · 3 评论 -
Pytorch在NLP中的简单应用
因为之前在项目中一直使用Tensorflow,最近需要处理NLP问题,对Pytorch框架还比较陌生,所以特地再学习一下pytorch在自然语言处理问题中的简单使用,这里做一个记录。一、Pytorch基础首先,第一步是导入pytorch的一系列包import torchimport torch.autograd as autograd #Autograd为Tensor...原创 2019-03-07 12:22:43 · 2232 阅读 · 0 评论 -
Pytorch的模型结构可视化(tensorboard)
在pytorch中,可以导入tensorboard模块,可视化网络结构及训练流程。下面通过“CNN训练MNIST手写数字分类”的小例子来学习一些可视化工具的用法,只需要加少量代码。一、tensorboardX的安装pip install tensorboardpip install tensorflowpip install tensorboardX二、导入tensorboa...原创 2019-03-12 00:19:50 · 11726 阅读 · 1 评论