
pytorch
lzrrrrr
死宅,dota2爱好者
展开
-
pytorch学习(6)pytorch中的数据加载DataLoader
文章目录pytorch学习(6)pytorch中的数据加载DataLoaderdataset数据加载顺序和采样器加载批处理和非批处理数据单进程和多进程数据加载 pytorch学习(6)pytorch中的数据加载DataLoader 明天肥家,这学期最后一篇博客啦 --------------------分割线------------------------------------ torch.ut...原创 2020-01-17 16:05:10 · 815 阅读 · 0 评论 -
pytorch学习(5)在MaskRCNN上进行finetuning
文章目录pytorch学习(5)在MaskRCNN上进行finetuning数据集准备定义我们的模型开始训练吧可视化 pytorch学习(5)在MaskRCNN上进行finetuning 在教程上正好看到一篇目标跟踪的教程,正好拿来练练手吧 这篇教程的目的是在MaskRCNN上进行微调来训练一个行人检测与分割模型 数据集准备 首先下载行人数据集,该数据的目录应该是这样的: annotation...原创 2020-01-06 21:18:41 · 1278 阅读 · 7 评论 -
pytorch学习(4)数据并行
pytorch学习(4)数据并行 pytorch60分钟教程最后一篇,这次学习如何使用DataParallel来调用多个GPU(然而实验室的GPU我还没用过) 在前面的学习中我们知道用如下方法来使用GPU: 把模型放入GPU中: device = torch.device("cuda:0") model.to(device) 把张量放入GPU中: mytensor = my_tensor.to(...原创 2020-01-05 15:07:17 · 262 阅读 · 0 评论 -
pytorch学习(3)训练一个图片分类器
文章目录pytorch学习(3)训练一个图片分类器用torchvision导入CIFAR10数据集并初始化定义网络结构定义损失函数和优化器训练测试 pytorch学习(3)训练一个图片分类器 与上一章不同,这次我们基于CIFAR10数据集训练一个完整的图片分类器,一般有如下步骤: 用torchvision导入CIFAR10数据集并初始化 torchvision是计算机视觉中很重要的软件包,其封装了...原创 2020-01-05 14:42:04 · 567 阅读 · 0 评论 -
pytorch学习(2)构造简单神经网络Lenet-5
文章目录pytorch学习(2)构造简单神经网络Lenet-5定义一个有训练参数的神经网络迭代的输入数据集计算LOSS反向传播梯度更新训练参数的值 pytorch学习(2)构造简单神经网络Lenet-5 用pytorch构造一个神经网络一般有以下几个步骤: 定义一个有训练参数的神经网络 import torch.nn as nn import torch.nn.functional as F c...原创 2020-01-05 12:48:33 · 215 阅读 · 0 评论 -
pytorch学习(1) Tensor与Autograd
文章目录pytorch学习笔记(1) Tensor与AutogradTensor创建tensortensor的操作Autograd pytorch学习笔记(1) Tensor与Autograd 开个专栏记录一些Pytorch的常用用法,方便自己查找。 一下内容均来自pytorch基础教程 Tensor tensor是每个深度学习框架都要用到的模块,在pytorch中,它有如下几种常见创建方法: 创...原创 2020-01-04 20:34:30 · 250 阅读 · 0 评论