鱼C论坛

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

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

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

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

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

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

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

  5.     val_set = torchvision.datasets.CIFAR10(root='./data', train=False,
  6.                                            download=True, transform=transform)
  7.     val_loader = torch.utils.data.DataLoader(val_set, batch_size=5000,
  8.                                              shuffle=False, num_workers=0)
复制代码


第二种方法:
  1. cifar10_dir = 'F:\Jupyter\CIFAR10'
  2. X_train, y_train, X_test, y_test = load_CIFAR10(cifar10_dir)
复制代码



我看大多数帖子都是用的第一种方法,但是搞不清这两种方法的区别,是不是因为第一种更好呢?球球大神帮忙解答一下
小甲鱼最新课程 -> https://ilovefishc.com
回复

使用道具 举报

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

本版积分规则

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

GMT+8, 2025-4-28 23:22

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

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