鱼C论坛

 找回密码
 立即注册
查看: 1678|回复: 0

请大神帮忙解决下这两段代码在划分数据集时候的使用区别

[复制链接]
发表于 2022-5-24 16:19:55 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
前提是我已经下载好了CIFAR10数据集。
第一种方法:
train_set = torchvision.datasets.CIFAR10(root='./data', train=True,
                                             download=False, transform=transform)

    train_loader = torch.utils.data.DataLoader(train_set, batch_size=36,
                                               shuffle=True, num_workers=0)

    val_set = torchvision.datasets.CIFAR10(root='./data', train=False,
                                           download=True, transform=transform)
    val_loader = torch.utils.data.DataLoader(val_set, batch_size=5000,
                                             shuffle=False, num_workers=0)

第二种方法:
cifar10_dir = 'F:\Jupyter\CIFAR10'
X_train, y_train, X_test, y_test = load_CIFAR10(cifar10_dir)


我看大多数帖子都是用的第一种方法,但是搞不清这两种方法的区别,是不是因为第一种更好呢?球球大神帮忙解答一下
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2024-11-18 03:21

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表