len(train

您所在的位置:网站首页 不丹国王豪车 len(train

len(train

#len(train| 来源: 网络整理| 查看: 265

这篇是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



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3