2.torch.utils.data
----
这个功能包的作用是收集、打包数据,给数据索引,然后按照 batch 将数据分批喂给神经网络。...、聚合成一个一个 batch 的自定义函数;
drop_last:一个布尔值,如果最后一批数据的个数不足 batch 的大小,是否保留这个 batch。...torch.utils.data.Subset(dataset, indices)
这个函数可以根据索引indices将数据集dataset分割。...([79, 60, 98, 74, 31, 43, 21, 69, 55, 76])
2.3. collate_fn 核对函数
----
这个变量的功能是在数据被读取后,送进模型前对所有数据进行处理、打包...将这个函数赋值给 collate_fn,在读取数据的时候就可以自动对数据进行 padding 并打包成一个 batch。
2.4 sampler 采样器
----
这个变量决定了数据读取的顺序。