python_F11 - 专业站长和开发者的学习网站

APP正在开发中...

返回顶部

python

在pytorch中为Module和Tensor指定GPU的例子

pytorch指定GPU 在用pytorch写CNN的时候，发现一运行程序就卡住，然后cpu占用率100%，nvidia-smi 查看显卡发现并没有使用GPU。所以考虑将模型和输

1480

2019-08-19

python
pytorch使用单个GPU与多个GPU进行训练与测试的教程

如下所示： device = torch.device(cuda:0 if torch.cuda.is_available() else cpu)#第一行代码model.to(device)#第二行代码首先是上面两行代码放在读取数据之前

1867

2019-08-19

python
将Pytorch模型从CPU转换成GPU的方法教程

最近将Pytorch程序迁移到GPU上去的一些工作和思考环境：Ubuntu 16.04.3 Python版本：3.5.2 Pytorch版本：0.4.0 0. 序言大家知道，在深度学习中使用

919

2019-08-19

python
pytorch GPU计算过程中出现内存耗尽的问题解决

Pytorch GPU运算过程中会出现：cuda runtime error(2): out of memory这样的错误。通常，这种错误是由于在循环中使用全局变量当做累加器，且累加梯度

1672

2019-08-19

python
python opencv实现证件照换底的方法

思路：先转到HSV空间，利用颜色提取背景制作掩模版mask，然后通过按位操作提取人像和制作新背景，最后叠加背景和人像得到换底后照片

1166

2019-08-19

python
pytorch 更改预训练模型网络结构的教程方法

一个继承nn.module的model它包含一个叫做children()的函数，这个函数可以用来提取出model每一层的网络结构，在此基础上进行修改即可，修改方法

1513

2019-08-19

python
pytorch打印网络结构的教程

最简单的方法当然可以直接print（net），但是这样网络比较复杂的时候效果不太好，看着比较乱；以前使用caffe的时候有一个网站可以在线生

951

2019-08-19

python
在PyTorch中Tensor的查找和筛选方法

本文源码基于版本1.0，交互界面基于0.4.1 import torch 按照指定轴上的坐标进行过滤 index_select() 沿着某tensor的一个轴dim筛选若干个坐标 x = tor

1109

2019-08-18

python
Pytorch Tensor的索引与切片实例

1. Pytorch风格的索引根据Tensor的shape，从前往后索引，依次在每个维度上做索引。示例代码： import torch a = torch.rand(4, 3, 28, 28)print(a[0].shape)

839

2019-08-18

python
PyTorch之图像和Tensor填充的实例

在PyTorch中可以对图像和Tensor进行填充，如常量值填充，镜像填充和复制填充等。在图像预处理阶段设置图像边界填充的方式如下： import v

1869

2019-08-18

python
Pytorch实现GoogLeNet的方法

GoogLeNet也叫InceptionNet，在2014年被提出，如今已到V4版本。GoogleNet比VGGNet具有更深的网络结构，一共有22层，但是参数比AlexNet要少12倍，但是计

1542

2019-08-18

python
解析PyTorch中nn.Linear的使用

Linear 的初始化部分： class Linear(Module): ... __constants__ = [bias] def __init__(self, in_features, out_features, bias=True): super(Linear, self).__init__() self.in_featur

1647

2019-08-18

python
PyTorch中Tensor的高阶操作解析

条件选取：torch.where(condition, x, y) Tensor 返回从 x 或 y 中选择元素的张量，取决于 condition 操作定义：举个例子： import torch c = randn(2, 3) cten

1642

2019-08-18

python
PyTorch中Tensor的拼接与拆分的实现

拼接张量：torch.cat() 、torch.stack() torch.cat(inputs, dimension=0) Tensor 在给定维度上对输入的张量序列 seq 进行连接操作举个例子： import torch x =

1469

2019-08-18

python
PyTorch中Tensor的维度变换实现方法

对于 PyTorch 的基本数据对象 Tensor (张量)，在处理问题时，需要经常改变数据的维度，以便于后期的计算和进一步处理，本文旨在列举一些维

1070

2019-08-18

python
Pytorch中的torch.gather函数的含义解析

pytorch中的gather函数 pytorch比tensorflow更加编程友好，所以准备用pytorch试着做最近要做的一些实验。立个flag开始学习pytorch，新开一个分类整理

1833

2019-08-18

python
pytorch索引查找 index_select的实例

index_select anchor_w = self.FloatTensor(self.scaled_anchors).index_select(1, self.LongTensor([0])) 参数说明：index_select(x, 1, indices) 1代表维度1，即列，indices是筛选

788

2019-08-18

python
pytorch 输出中间层特征的示例

pytorch 输出中间层特征： tensorflow输出中间特征，2种方式： 1. 保存全部模型(包括结构)时，需要之前先add_to_collection 或者用slim模块下的end

1314

2019-08-18

python
pytorch 模型可视化的实例

如下所示：一. visualize.py from graphviz import Digraphimport torchfrom torch.autograd import Variable def make_dot(var, params=None): Produces Graphviz representation of PyT

569

2019-08-18

python