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

你可能感兴趣的:(PyTorch学习之路,深度学习)