
pytorch
代码的路
公众号:代码的路
编程技术分享,包括深度学习、Python、Java等技术
展开
-
PyTorch的简单实现
1.必要的 PyTorch 背景PyTorch 是一个建立在 Torch 库之上的 Python 包,旨在加速深度学习应用。 PyTorch 提供一种类似 NumPy 的抽象方法来表征张量(或多维数组),它可以利用 GPU 来加速训练。1.1 PyTorch 张量PyTorch 的关键数据结构是张量,即多维数组。其功能与 NumPy 的 ndarray 对象类似,如下我们可以使用 ...原创 2019-03-20 19:13:09 · 2898 阅读 · 1 评论 -
PyTorch中 nn.Conv2d与nn.ConvTranspose2d函数的用法
1. 通道数问题:描述一个像素点,如果是灰度,那么只需要一个数值来描述它,就是单通道。如果有RGB三种颜色来描述它,就是三通道。最初输入的图片样本的channels,取决于图片类型; 卷积操作完成后输出的out_channels,取决于卷积核的数量。此时的out_channels也会作为下一次卷积时的卷积核的in_channels; 卷积核中的in_channels,上...原创 2019-02-23 16:13:24 · 50318 阅读 · 4 评论 -
Pytorch中 nn.BatchNorm2d() 归一化操作
官方文档:原创 2019-05-21 21:27:35 · 3882 阅读 · 0 评论 -
PyTorch中 torch.nn与torch.nn.functional的区别
torch.nnpytorch中文文档:torch.nn在__init__()函数里定义,定义的是一个类:torch.nn.functionalpytorch中文文档:torch.nn.functional在__forward()__函数里定义,定义的是一个函数:两者的区别:nn中是定义的类,以class xx来定义的,可以提取变化的学习参数。nn.functional...原创 2019-05-22 17:40:32 · 5935 阅读 · 1 评论 -
ImportError: cannot import name ‘imresize‘
出现错误:ImportError: cannot import name 'imresize'1.安装pillow:pip install pillow2.安装早期版本:scipy版本问题,高级版本不再包含函数imresize,官网给出的解释及解决办法如下:imresize is deprecated! imresize is deprecated in SciPy 1.0...原创 2019-06-06 11:26:26 · 4535 阅读 · 0 评论 -
pytorch指定GPU
1直接终端中设定:CUDA_VISIBLE_DEVICES=1 python main.py2 python代码中设定:import osos.environ["CUDA_VISIBLE_DEVICES"] = "0,1"原创 2019-06-03 15:45:50 · 359 阅读 · 0 评论