torchvision.datasets.CIFAR10()和torch.utils.data.DataLoader详细介绍

Amanda_ABAP_Python

于 2021-01-30 14:45:02 发布

阅读量3k

点赞数 10

分类专栏： python 数据分析人工智能文章标签： python 深度学习

本文链接：https://blog.csdn.net/Amanda_python/article/details/113430017

版权

import torchvision  

trainset = torchvision.datasets.CIFAR10(root = data_path,
                                        train = True,
                                        download = True,
                                        transform=transform)
'''                                        
root : cifar-10 的根目录，data_path（路径）
train : True = 训练集, False = 测试集
download : True

最低0.47元/天解锁文章