pytorch
文章平均质量分 91
Mason_Luo_19
这个作者很懒,什么都没留下…
展开
-
常用工具
常用工具1 数据加载和预处理import torchtorch.__version__'1.3.1'1.1 DatasetDataset是一个抽象类,为方便读取,需将要使用的数据包装为Dataset类。自定义的Dataset需要继承它并实现两个成员方法:getitem() 用索引(0到len(self))获取一条数据或一个样本。obj[index]等价于obj.getitem...原创 2019-12-02 16:06:04 · 163 阅读 · 0 评论 -
训练分类器
%matplotlib inline关于数据?一般情况下处理图像、文本、音频、视频时,可以用标准的python包加载数据到一个numpy数组中,然后将数组转换为torch.Tensor。图像使用Pillow、OpenCV音频使用scipy、librosa文本使用python、cython特别的,对于图像任务,使用torchvision包处理一下基本图像数据集。如使用torchv...原创 2019-12-02 16:05:06 · 712 阅读 · 0 评论 -
神经网络
使用torch.nn构建神经网络一个nn.Module包含各个层和一个forward(input)方法,该方法返回output。更新网络参数:weight = weight - learning_rate * gradient定义网络import torchimport torch.nn as nnimport torch.nn.functional as Fclass Net...原创 2019-12-02 16:03:27 · 189 阅读 · 0 评论 -
Autograd:自动求导
import torch# 创建张量并设置requires_grad=True来追踪其计算历史x = torch.ones(2, 2, requires_grad=True)print(x)tensor([[1., 1.], [1., 1.]], requires_grad=True)y = x + 2print(y)tensor([[3., 3.], ...原创 2019-12-02 16:02:56 · 160 阅读 · 0 评论 -
pytorch张量
from __future__ import print_functionimport torchimport numpy as nptorch.__version__'1.3.1'x = torch.empty(5,3)print(x)tensor([[9.1834e-41, 0.0000e+00, 0.0000e+00], [0.0000e+00, 0.00...原创 2019-12-02 16:01:02 · 222 阅读 · 0 评论