Tra_cy的博客

使用背景：模型有3个输出，需要对这三个输出加权求和，于是设置可学习的参数作为权重，代码如下：self.w1 = torch.nn.Parameter(torch.FloatTensor(1), requires_grad=True)self.w2 = torch.nn.Parameter(torch.FloatTensor(1), requires_grad=True)self.w3 = t...

实际应用时可能比较想获取VGG中间层的输出，那么就可以如下操作：import numpy as npimport torchfrom torchvision import modelsfrom torch.autograd import Variableimport torchvision.transforms as transforms class CNNShow(): d...

pytorch中使用interpolate可以很轻松的完成上采样和下采样。官方文档torch.nn.functional.interpolate(input, size=None, scale_factor=None, mode=‘nearest’, align_corners=None)函数的参数如下：input (Tensor) – the input tensorsize (...

文章目录安装Anaconda安装PyTorch安装Anaconda下载：https://repo.continuum.io/archive/下载相应版本的Anaconda，这里使用的是版本。安装：双击安装包，基本上都是“下一步”，注意选择安装路径和加入环境变量即可。创建环境：右击Anaconda Prompt，选择以管理员身份运行conda创建环境命令为：conda create -...

举一个很简单的例子来熟悉使用pytorch构建和训练神经网络的整个过程。import torchimport torch.nn as nnN, D_in, H, D_out = 64, 1000, 100, 10#随机创建一些训练数据x = torch.randn(N, D_in)y = torch.randn(N, D_out)#定义modelclass...

文章目录前言DatasetDataLoader前言Pytorch 中比较重要的是对数据的处理，其中，进行数据读取的一般有三个类：DatasetDataLoaderDataLoaderIter其中，这三是一个依次封装的关系：“Dataset被封装进DataLoader，DataLoader再被封装进DataLoaderIter”Datasetclass torch.utils.d...

基于PyTorch的一些预训练模型下载地址：https://github.com/pytorch/vision/tree/master/torchvision/models

参考package的使用方法：https://pytorch-cn.readthedocs.io/zh/latest/神经网络可以通过 torch.nn 包来构建。一个典型的神经网络训练过程包括以下：定义一个包含可训练参数的神经网络迭代整个输入通过神经网络处理输入计算损失loss反向传播梯度到神经网络的参数更新网络的参数，典型的更新方法是：weight = weight - lea...

文章目录Tensors（张量）Tensors（张量）Tensors其实就是多维数组，Tensors类似于NumPy的ndarrays，同时Tensors可以使用GPU进行计算。构造一个5*3矩阵import torchx = torch.empty(5,3) #不初始化x = torch.rand(5,3) #随机初始化x = torch.zeros(5,3,dtype=t...

文章目录pytorch简介pytorch安装pytorch简介PyTorch 是一个基于 Torch（LUA语言） 的 Python 开源机器学习库，用于自然语言处理等应用程序。 它主要由 Facebook 的人工智能研究小组开发。PyTorch是一个Python包，提供两个高级功能：具有强大的GPU加速的张量计算（如NumPy）包含自动求导系统的的深度神经网络pytorch安装在...