浅析PyTorch中nn.Module的使用
yipeiwu_com6年前
torch.nn.Modules 相当于是对网络某种层的封装，包括网络结构以及网络参数和一些操作 torch.nn.Module 是所有神经网络单元的基类 查看源码 初始化部分：...
在pytorch中查看可训练参数的例子
yipeiwu_com6年前
pytorch中我们有时候可能需要设定某些变量是参与训练的，这时候就需要查看哪些是可训练参数，以确定这些设置是成功的。 pytorch中model.parameters()函数定义如下：...
Pytorch加载部分预训练模型的参数实例
yipeiwu_com6年前
前言 自从从深度学习框架caffe转到Pytorch之后，感觉Pytorch的优点妙不可言，各种设计简洁，方便研究网络结构修改，容易上手，比TensorFlow的臃肿好多了。对于深度学习...
pytorch中的embedding词向量的使用方法
yipeiwu_com6年前
Embedding 词嵌入在 pytorch 中非常简单，只需要调用 torch.nn.Embedding(m, n) 就可以了，m 表示单词的总数目，n 表示词嵌入的维度，其实词嵌入就...
对Pytorch神经网络初始化kaiming分布详解
yipeiwu_com6年前
函数的增益值 torch.nn.init.calculate_gain(nonlinearity, param=None) 提供了对非线性函数增益值的计算。 增益值gain是一个...
在PyTorch中Tensor的查找和筛选例子
yipeiwu_com6年前
本文源码基于版本1.0，交互界面基于0.4.1 import torch 按照指定轴上的坐标进行过滤 index_select() 沿着某tensor的一个轴dim筛选若干个坐标 &...
Pytorch Tensor的索引与切片例子
yipeiwu_com6年前
1. Pytorch风格的索引 根据Tensor的shape，从前往后索引，依次在每个维度上做索引。 示例代码： import torch a = torch.rand(4, 3...
PyTorch之图像和Tensor填充的实例
yipeiwu_com6年前
在PyTorch中可以对图像和Tensor进行填充，如常量值填充，镜像填充和复制填充等。在图像预处理阶段设置图像边界填充的方式如下： import vision.torchvisio...
Pytorch实现GoogLeNet的方法
yipeiwu_com6年前
GoogLeNet也叫InceptionNet，在2014年被提出，如今已到V4版本。GoogleNet比VGGNet具有更深的网络结构，一共有22层，但是参数比AlexNet要少12倍...
浅析PyTorch中nn.Linear的使用
yipeiwu_com6年前
查看源码 Linear 的初始化部分： class Linear(Module): ... __constants__ = ['bias'] def __init__(s...
