pytorch
关注
分享
扫一扫
文章平均质量分 55
蓝鲸123
做更好的自己
展开
-
0.4.0pytorch运行过程中对0-dim和volatile提示UserWarning的解决方法
将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告，它在提醒用户下个版本这将成为一个错误UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python...转载 2018-06-23 13:25:25 · 2605 阅读 · 1 评论 -
PyTorch 常用方法总结4：张量维度操作（拼接、维度扩展、压缩、转置、重复……）
涉及的方法有下面几种torch.cat() torch.Tensor.expand()torch.squeeze() torch.Tensor.repeat()torch.Tensor.narrow() torch.Tensor.view()torch.Tensor.resize_() torch.Tensor.permute()拼接张量torch.cat(seq, dim=0...原创 2018-10-16 15:09:56 · 56356 阅读 · 6 评论 -
使用tuna等源安装pytorch和tensorflow
阿里云或者豆瓣的镜像。阿里云 http://mirrors.aliyun.com/pypi/simple/中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/豆瓣 http://pypi.douban.com/simple/清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/中国科学技术大学 http://...原创 2018-09-21 17:07:52 · 2836 阅读 · 0 评论 -
grad_fn 里面的next_functions 属性
next_functions 反向传播计算图的记录。 在反向图的计算中，计算图终止于叶子AccumulateGrad节点。有一个.variable属性指向叶子节点。例子：a = torch.randn(1, requires_grad=True)b = a*(a+2)print (b.grad_fn.next_functions)print (b.grad_fn.next_fu...原创 2018-09-05 10:31:29 · 6296 阅读 · 1 评论 -
Pytorch常用函数解析（一） Tensor 拼接
torch模块下的数学操作符1 . torch.numel() 返回一个tensor变量内所有元素个数，可以理解为矩阵内元素的个数2 . torch.squeeze() 对于tensor变量进行维度压缩，去除维数为1的的维度。例如一矩阵维度为A*1*B*C*1*D，通过squeeze()返回向量的维度为A*B*C*D。squeeze(a)，表示将a的维数位1的维度删掉，squeeze(a,...原创 2018-08-23 13:30:21 · 16107 阅读 · 1 评论 -
Pytorch变量类型转换
Pytorch的数据类型为各式各样的Tensor,Tensor可以理解为高维矩阵。与Numpy中的Array类似。Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型，一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。通过使用Type函数可以查看变量类型。一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(...转载 2018-08-20 14:10:55 · 1380 阅读 · 0 评论 -
pytorch使用多GPU训练MNIST
下面的代码参数没有调试，可能准确率不高，仅仅供参考代码格式。import argparseimport torchimport torch.nn as nnimport torch.optim as optimimport torch.nn.functional as Ffrom torchvision import datasetsfrom torchvision impo...原创 2018-08-20 11:40:53 · 2316 阅读 · 0 评论 -
tf.contrib.keras.preprocessing.sequence.pad_sequences 将标量数据 转换成numpy ndarray
keras.preprocessing.sequence.pad_sequences(sequences, maxlen=None, dtype=’int32’, padding=’pre’, truncating=’pre’, value=0.) 函数说明： 将长为nb_samples的序列（标量序列）转化为形如(nb_samples,nb_timesteps)2D numpy arr...原创 2018-07-03 21:50:53 · 6688 阅读 · 0 评论 -
pytorch sampler对数据进行采样
PyTorch中还单独提供了一个sampler模块，用来对数据进行采样。常用的有随机采样器：RandomSampler，当dataloader的shuffle参数为True时，系统会自动调用这个采样器，实现打乱数据。默认的是采用SequentialSampler，它会按顺序一个一个进行采样。这里介绍另外一个很有用的采样方法： WeightedRandomSampler，它会根据每个样本的权重选取数...原创 2018-07-01 21:32:31 · 35575 阅读 · 6 评论 -
pytorch 函数DataLoader
Dataset（https://blog.csdn.net/TH_NUM/article/details/80877196）只负责数据的抽象，一次调用getitem只返回一个样本。前面提到过，在训练神经网络时，最好是对一个batch的数据进行操作，同时还需要对数据进行shuffle和并行加速等。对此，PyTorch提供了DataLoader帮助我们实现这些功能。DataLoader的函数定义如...转载 2018-07-01 21:23:34 · 51958 阅读 · 1 评论 -
pytorch Dataset 的ImageFolder
介绍常用的pytorch Dataset 里面的ImageFolder，实现和https://blog.csdn.net/TH_NUM/article/details/80877196很相似。 ImageFolder假设所有的文件按文件夹保存好，每个文件夹下面存贮同一类别的图片，文件夹的名字为分类的名字。ImageFolder(root,transform=None,target_tr...原创 2018-07-01 20:57:11 · 42343 阅读 · 4 评论 -
pytorch 数据处理：定义自己的数据集合
数据处理版本1#数据处理import osimport torchfrom torch.utils import datafrom PIL import Imageimport numpy as np#定义自己的数据集合class DogCat(data.Dataset): def __init__(self,root): #所有图片的绝对路径...原创 2018-07-01 20:33:02 · 12139 阅读 · 1 评论 -
pytorch 函数clamp
clamp表示夹紧，夹住的意思，torch.clamp(input,min,max,out=None)-> Tensor将input中的元素限制在[min,max]范围内并返回一个Tensor用法：原创 2018-06-29 18:19:55 · 21507 阅读 · 0 评论 -
pytorch 常用函数 max ,eq
max找出tensor 的行或者列最大的值：找出每行的最大值:import torchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,1))输出： (tensor([ 1., 2., 3.]), tensor([ 0, 0, 0]))找出每列的最大值：import ...原创 2018-06-23 13:56:53 · 25513 阅读 · 1 评论 -
pytorch Distribute分布式训练
from torch.multiprocessing import Processfrom torch.utils.data import Dataset, DataLoaderimport numpy as npimport osimport torchimport torch.distributed as distimport torch.nn as nnimport torch...原创 2018-11-21 10:37:25 · 3187 阅读 · 1 评论