我们经常可以看到Pytorch加载数据集会用到官方整理好的数据集。...很多时候我们需要加载自己的数据集,这时候我们需要使用Dataset和DataLoader
Dataset:是被封装进DataLoader里,实现该方法封装自己的数据和标签。...np.random.rand(10, 20)
# 随机生成标签,大小为10 * 1列
source_label = np.random.randint(0,2,(10, 1))
# 通过GetLoader将数据进行加载...,batch_size,shuffle,drop_last,num_workers)
参数含义如下:
d a t a s e t \color{HotPink}{dataset} dataset: 加载..., batch_size=6, shuffle=True, drop_last=False, num_workers=2)
此时,我们的数据已经加载完毕了,只需要在训练过程中使用即可。