Pytorch DataLoader shuffle 参数源码解读
DataLoader 的使用方法:调用 dataloader. __iter__ 获取迭代器调用 迭代器的 __next__ 获取下一个 batch首先 dataloader 可以设置是否 shuffle那么只要看 shuffle 参数对这个过程有什么影响即可class DataLoader(Generic[T_co]):def __init__(self, dataset: Dataset[T_co], batch_size: Optional[int] = 1,
原创
2021-08-05 14:08:03 ·
4241 阅读 ·
0 评论