1. CIFAR10 数据集
CIFAR10 是深度学习入门最先接触到的数据集之一,主要用于图像分类任务中,该数据集总共有10个
类别。
- 图片数量:
6w 张
- 图片宽高:
32x32
- 图片类别:10
- Trainset: 5w 张,5 个训练块
- Testset: 1w 张,1 个测试块
Pytorch 集成了很多常见数据集的API, 可以通过pytorch 来下载这些数据集,并做成模型需要的DataSet和Dataloader, 供模型训练使用。文章来源:https://www.toymoban.com/news/detail-673117.html
1.1 数据集准备
制作Dataloader: utils.py
代码实现如下:文章来源地址https://www.toymoban.com/news/detail-673117.html
import torchvision.datasets as datasets
import torchvision.transforms as transforms
到了这里,关于剪枝基础与实战(2): L1和L2正则化及BatchNormalization讲解的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!