len(train |
您所在的位置:网站首页 › 不丹国王豪车 › len(train |
这篇是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 |