train_dataloader = DataLoader(dataset=dataset,batch_size=4,sampler=sampler.SequentialSampler(range(24)))
print(len(train_dataloader)
pytorch 用sampler划分数据集
最新推荐文章于 2023-06-20 13:13:07 发布
本文介绍了如何使用PyTorch的DataLoader进行数据加载,通过设置sampler参数,创建了训练集和验证集的数据加载器。训练集DataLoader包含前24个样本,每个batch大小为4,总批次数为6。验证集DataLoader则包含剩余的4个样本,批次数为1。这展示了在深度学习中如何有效地划分并处理数据。
摘要由CSDN通过智能技术生成