DataLoader(dataset, batch_size=1, shuffle=False, sampler=None,
batch_sampler=None, num_workers=0, collate_fn=None,
pin_memory=False, drop_last=False, timeout=0,
worker_init_fn=None, *, prefetch_factor=2,
persistent_workers=False)
- dataset (Dataset): 提供数据的数据集对象。
- batch_size (int, optional): 每个批次中包含的样本数。默认为 1。
- shuffle (bool, optional): 是否在每个 epoch 开始时打乱数据。默认为 False。
- sampler (Sampler, optional): 定义从数据集中抽取样本的策略。如果指定,shuffle 必须为 False。
- batch_sampler (Sampler, optional): 与 sampler 类似,但直接返回批次的索引,而不是单个样本的索引。如果指定,batch_size, shuffle, sampler, 和 drop_last 必须不被指定。
- num_workers (int, optional): 用于数据加载的子进程数。0 表示数据将在主进程中加载。默认为 0。
- drop_last (bool, optional): 如果数据集大小不能被 batch_size 整除,设置为 True 后将丢弃最后一个不完整的批次。默认为 False。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· AI技术革命,工作效率10个最佳AI工具