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