PyTorch
my-GRIT
Keep living or Keep dying
展开
-
PyTorch中使用指定的GPU(转)
点击打开链接转载自:http://www.cnblogs.com/darkknightzh/p/6836568.htmlPyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。有如下两种方法来指定需要使用的GPU。1. 类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。1.1 直接终端中设定:CUDA_VISIBLE_DEVICE...转载 2018-06-17 21:13:55 · 541 阅读 · 0 评论 -
PyTorch之Variable的detach和detach_
本文源自:https://blog.csdn.net/u012436149/article/details/76714349这里加入自己的一点实践。pytorch 的 Variable 对象中有两个方法,detach和 detach_ 本文主要介绍这两个方法的效果和 能用这两个方法干什么。detach官方文档中,对这个方法是这么介绍的。返回一个新的 从当前图中分离的 Varia...转载 2018-08-20 10:23:57 · 1664 阅读 · 0 评论 -
PyTorch之Tensor
之前看过tensor的操作,不过对tensor中的squeeze和unsqueeze不太明白,今天看到一篇博客,理解更深了些,这里只截取有关squeeze和unsqueeze的部分。本文转自:https://blog.csdn.net/zyn0318/article/details/802993791. torch.squeeze(input, dim = None, out = None...转载 2018-08-19 21:18:02 · 1058 阅读 · 0 评论 -
PyTorch之transforms
本文转自:https://zhuanlan.zhihu.com/p/27382990为了方便进行数据的操作,pytorch团队提供了一个torchvision.transforms包,我们可以用transforms进行以下操作:PIL.Image/numpy.ndarray与Tensor的相互转化; 归一化,即0均值,1方差; 对PIL.Image进行裁剪、缩放等操作。 通常...转载 2018-08-19 14:56:51 · 963 阅读 · 0 评论 -
BatchNormalization
文章转自:https://blog.csdn.net/hjimce/article/details/50866313一、背景意义本篇博文主要讲解2015年深度学习领域,非常值得学习的一篇文献:《Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift》,这个...转载 2018-08-18 20:17:16 · 269 阅读 · 0 评论 -
Ubuntu 16.04+pytorch(cpu版)
这是我参考的原文,只是个别地方添加或修改。https://blog.csdn.net/tuyim7124/article/details/80723997一、目标环境ubuntu16.04+python3.6+pytorch(CPU版)二、安装步骤(一)安装anaconda1.在 anaconda官网 选择 Python 3.6 version 下载,考虑到服务器在国外,下...转载 2018-08-04 16:22:00 · 4444 阅读 · 1 评论 -
Pytorch入门学习(九)---detach()的作用(从GAN代码分析)
转自:点击打开链接总说简单来说detach就是截断反向传播的梯度流。 def detach(self): """Returns a new Variable, detached from the current graph. Result will never require gradient. If the input is volatile, the o...转载 2018-06-25 16:50:29 · 33336 阅读 · 2 评论 -
PyTorch的torch.cat
1. 字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起。2. 例子理解>>> import torch>>> A=torch.ones(2,3) #2x3的张量(矩阵) >>> Ate...原创 2018-06-25 15:55:59 · 276168 阅读 · 34 评论 -
pytorch 0.4.0升级问题
转自点击打开链接pytorch 0.3.0升级到0.4.0之后部分之前保存的模型有些不能直接使用判断版本:import torchprint(torch.__version__)0.4.0相比0.3.0代码中需要修改的部分:1.norm_layernorm_layer = functools.partial(nn.InstanceNorm2d, affine=False)改为:norm_laye...转载 2018-06-18 10:46:57 · 1214 阅读 · 0 评论 -
Pytorch 之修改Tensor部分值
一:背景引入对于一张图片,怎样修改局部像素值?二:利用Tensor方法比如输入全零tensor,可认为为黑色图片>>> n=torch.FloatTensor(3,3,4).fill_(0)>>> n tensor([[[0., 0., 0., 0.], [0., 0., 0., 0.]...原创 2018-09-16 10:07:57 · 31219 阅读 · 0 评论