len(train_loader),len(label)和训练数据集个数之间的关系

这篇是PyTorch学习之路的第六篇

问题

想要知道len(train_loader),len(label)和训练数据集个数之间的关系

解答

以50000张图像的训练集为例,
若设train_loader = DataLoader(train_dataset,batch_size=4,shuffle=True,num_workers=num_workers)

则相当于将训练集平均分成12500份,每份有4张图片(batch_size参数设置的就是每份中有多少张图片)

train_loader中的每个元素相当于一个分组,一个组中有4张图片,label就是一个分组中的一张图片的标签

故len(train_loader)==12500,len(label)==4

  • 52
    点赞
  • 72
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值