0

0

如何在 Hugging Face 中按需加载数据集子集以节省磁盘空间

聖光之護

聖光之護

发布时间:2026-01-31 21:02:03

|

631人浏览过

|

来源于php中文网

原创

如何在 Hugging Face 中按需加载数据集子集以节省磁盘空间

hugging face 的 `datasets` 库默认会完整下载整个数据集,即使使用切片语法(如 `"train[:20%]"`)也无法避免;要真正实现轻量级加载,必须启用流式(streaming)模式,它支持边迭代边读取、无需本地存储全量数据。

在微调如 facebook/wav2vec2 等大型语音模型时,常需使用 Common Voice 等超大规模数据集(例如 common_voice_16_0 达 256GB),但本地磁盘空间有限。此时若误用常规切片方式:

from datasets import load_dataset
# ❌ 错误:仍会完整下载并解压全部数据(约256GB)
ds = load_dataset("mozilla-foundation/common_voice_16_0", "en", split="train[:20%]")

该写法不会跳过下载——split="train[:20%]" 仅在数据已全部加载进内存/磁盘后才执行逻辑切片,因此无法缓解磁盘压力。

✅ 正确方案是启用 streaming=True
它绕过本地缓存与完整下载,直接通过网络流式拉取样本,配合 .take()、.skip() 或 .shard() 等方法实现真正的“按需获取”:

from datasets import load_dataset

# ✅ 启用 streaming:不下载任何文件到磁盘
dataset_name = "mozilla-foundation/common_voice_16_0"
ds = load_dataset(dataset_name, "en", split="train", streaming=True)

# 取前 100 条样本(适用于快速验证或小规模实验)
ds_subset = ds.take(100)

# 转为可迭代对象(注意:streaming 数据集不支持索引 ds[0])
for example in ds_subset:
    print(example["sentence"])  # 示例字段
    break

? 关键注意事项

琅琅配音
琅琅配音

全能AI配音神器

下载
  • 流式数据集返回的是 IterableDataset,不支持随机访问(如 ds[42] 会报错),仅支持顺序遍历;
  • 若需打乱(shuffle),须显式调用 .shuffle(buffer_size=2000),其中 buffer_size 决定内存中缓存样本数(不宜过大);
  • 训练时建议搭配 torch.utils.data.IterableDataset 或 datasets.IterableDataset.to_torch_dataset() 使用;
  • 部分数据集(尤其是含大音频/视频文件的)可能因服务器限速或网络波动影响流式稳定性,建议添加重试逻辑或预缓存关键子集。

? 进阶技巧
若需固定比例采样(如 20%),可用 .shard() + .skip() 组合模拟,或借助 itertools.islice:

import itertools
ds = load_dataset("mozilla-foundation/common_voice_16_0", "en", split="train", streaming=True)
ds_20_percent = itertools.islice(ds, 0, None, 5)  # 每5条取1条 ≈ 20%,适合均匀采样

总之,streaming=True 是应对海量数据集磁盘瓶颈的核心机制——它将“下载→加载→切片”三步简化为“按需流式拉取”,兼顾效率与资源可控性。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
go语言 数组和切片
go语言 数组和切片

本专题整合了go语言数组和切片的区别与含义,阅读专题下面的文章了解更多详细内容。

57

2025.09.03

vscode 格式化
vscode 格式化

本专题整合了vscode格式化相关内容,阅读专题下面的文章了解更多详细内容。

0

2026.03.18

vscode设置中文教程
vscode设置中文教程

本专题整合了vscode设置中文相关内容,阅读专题下面的文章了解更多详细教程。

0

2026.03.18

vscode更新教程合集
vscode更新教程合集

本专题整合了vscode更新相关内容,阅读专题下面的文章了解更多详细教程。

2

2026.03.18

Gemini网页版零基础入门:5分钟上手Gemini聊天指南
Gemini网页版零基础入门:5分钟上手Gemini聊天指南

本专题专为零基础用户打造,5分钟快速掌握Gemini网页版核心用法。从账号登录到界面布局,详解如何发起对话、优化提示词及利用多模态功能。通过实战案例,教你高效获取信息、创作内容与分析数据。无论学习还是工作,轻松开启AI辅助新时代,让Gemini成为你的得力智能助手。

4

2026.03.18

Python WebSocket实时通信与异步服务开发实践
Python WebSocket实时通信与异步服务开发实践

本专题聚焦 Python 在实时通信场景中的开发实践,系统讲解 WebSocket 协议原理、长连接管理、消息推送机制以及异步服务架构设计。内容包括客户端与服务端通信实现、连接稳定性优化、消息队列集成及高并发处理策略。通过完整案例,帮助开发者构建高效稳定的实时通信系统,适用于聊天应用、实时数据推送等场景。

8

2026.03.18

Java Spring Security权限控制与认证机制实战
Java Spring Security权限控制与认证机制实战

本专题围绕 Java 后端安全体系建设展开,重点讲解 Spring Security 在权限控制与认证机制中的应用实践。内容涵盖用户认证流程、权限模型设计、JWT 鉴权方案、OAuth2 集成以及接口安全防护策略。通过实际项目案例,帮助开发者构建安全可靠的后端认证体系,提升系统安全性与可扩展能力。

22

2026.03.18

抖漫入口地址合集
抖漫入口地址合集

本专题整合了抖漫入口地址相关合集,阅读专题下面的文章了解更多详细地址。

169

2026.03.17

多环境下的 Nginx 安装、结构与运维实战
多环境下的 Nginx 安装、结构与运维实战

本专题聚焦多环境下Nginx实战,详解开发、测试及生产环境的差异化安装策略与目录结构规划。深入剖析配置模块化设计、灰度发布流程及跨环境同步机制。结合监控告警、故障排查与自动化运维工具,提供全链路管理方案,助力团队构建灵活、高可用的Nginx服务体系,从容应对复杂业务场景挑战。

16

2026.03.17

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
React 教程
React 教程

共58课时 | 6.2万人学习

React中文开发手册
React中文开发手册

共0课时 | 1人学习

react快速入门
react快速入门

共14课时 | 2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号