pytorch学习笔记(六):自定义Datasets
来源:互联网 发布:反渗透 纯水机 知乎 编辑:程序博客网 时间:2024/05/26 20:23
自定义Datasets
什么是Datasets:
在输入流水线中,我们看到准备数据的代码是这么写的data = datasets.CIFAR10("./data/", transform=transform, train=True, download=True)
。datasets.CIFAR10
就是一个Datasets
子类,data
是这个类的一个实例。
为什么要定义Datasets:
PyTorch
提供了一个工具函数torch.utils.data.DataLoader
。通过这个类,我们在准备mini-batch
的时候可以多线程并行处理,这样可以加快准备数据的速度。Datasets
就是构建这个类的实例的参数之一。
如何自定义Datasets
下面是一个自定义Datasets的框架
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
下面看一下官方MNIST
的例子(代码被缩减,只留下了重要的部分):
阅读全文
0 0
- pytorch学习笔记(六):自定义Datasets
- pytorch学习笔记(六):自定义Datasets
- Pytorch学习笔记(六)
- pytorch学习笔记(1)--pytorch张量
- pytorch 学习笔记之自定义 Module
- Pytorch 学习笔记之自定义 Module
- Pytorch学习笔记(一)
- Pytorch学习笔记(二)
- Pytorch学习笔记(三)
- Pytorch学习笔记(四)
- Pytorch学习笔记(五)
- PyTorch代码学习-torchvision.datasets中folder.py
- pytorch学习笔记(九):PyTorch结构介绍
- pytorch学习笔记(九):PyTorch结构介绍
- Pytorch学习笔记(一):pytorch的安装-Ubuntu14.04
- pytorch学习笔记(十七):python 端扩展 pytorch
- pytorch学习笔记(十八):C 语言扩展 pytorch
- pytorch学习笔记(七):pytorch hook 和 关于pytorch backward过程的理解
- Servlet简介
- pytorch学习笔记(五):保存和加载模型
- 【代码积累】TCP client
- 外企入职记第一回
- Seaborn绘图
- pytorch学习笔记(六):自定义Datasets
- 分布式模块化 Java 开发平台 Castle-Platform
- 【代码积累】TCP server
- 关于Shell编程的几个典型企业面试题
- MUI 的侧滑菜单 详解
- IPC
- pytorch学习笔记(七):pytorch hook 和 关于pytorch backward过程的理解
- Android TabLayout添加自定义分割线并且可以修改分割线高度
- 现代开发人员必备的十款工具