讲师中心 微信公众号
AI工具推荐 视频效率加速

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

小伟大大_2231

小伟大大_2231

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

|

651人浏览过

|

来源于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%]" 仅在数据已全部加载进内存/磁盘后才执行逻辑切片,因此无法缓解磁盘压力。

Facebook Page Manager 1.0.0
Facebook Page Manager 1.0.0

通过 Meta Graph API 管理 Facebook 页面。发布内容(文字、图片、链接),列出帖子,管理评论(列出/回复/隐藏/删除),适用于用户想要发布...

下载

✅ 正确方案是启用 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

? 关键注意事项:

  • 流式数据集返回的是 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工具

更多
音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

墨刀AI
墨刀AI Hot

一款AI图像与设计工具,主要用于产品经理的专属智能体,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

AionClaw
AionClaw Hot

AionClaw是一款面向办公、创作和编程任务的AI桌面智能体。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

相关专题

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

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

1612

2025.09.03

FrankenPHP集成Laravel详细教程
FrankenPHP集成Laravel详细教程

本专题提供FrankenPHP集成Laravel的详细配置指南,全面解析运行原理、开发环境搭建、Caddyfile配置、Octane工作模式、数据库连接、队列任务、定时任务和生产环境优化,解决部署过程中常见的报错与兼容性问题。

0

2026.10.08

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

120

2026.09.30

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

100

2026.09.30

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

80

2026.09.30

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

80

2026.09.30

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

100

2026.09.29

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

300

2026.09.23

Buffalo框架路由与请求处理实操指南
Buffalo框架路由与请求处理实操指南

本专题讲解Buffalo框架路由与请求处理机制,涵盖路由注册与分组、资源路由、Handler编写规范、Context上下文方法、参数绑定、中间件编写挂载、Session与Cookie读写、Flash消息及错误页面定制方法。

180

2026.09.23

热门下载

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

精品课程

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

共58课时 | 12.1万人学习

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

共0课时 | 1人学习

react快速入门
react快速入门

共14课时 | 2.3万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn