使用HuggingFace Datasets加载AMI数据集出现AttributeError如何解决
错误原因
你遇到的报错是本地安装的HuggingFace Datasets库版本过旧,和远程主分支的AMI数据集配置不兼容导致的:
- 从报错信息可以看出你当前使用的是1.8.0版本的Datasets库,该版本的官方仓库中没有收录AMI数据集的配置文件,因此程序自动拉取了主分支上的最新配置
- 新版本的AMI数据集配置用到了旧版Datasets库不支持的任务模板,导致
NAME2TEMPLATE.get(task_name)返回空值,调用from_dict方法时触发属性错误
解决方法
你可以选择以下任意一种方案解决问题:
- 升级Datasets库到最新版本,这是最稳妥的解决方案,执行命令:
升级完成后重新运行原加载代码即可正常加载数据集pip install --upgrade datasets - 如果你因环境限制无法升级依赖,可以在加载数据集时指定和你当前库版本匹配的配置版本,避免拉取不兼容的最新配置:
注:如果指定版本后提示找不到对应数据集,说明1.8.0版本确实未收录AMI数据集,仍需升级Datasets库后使用。from datasets import load_dataset, load_metric dataset = load_dataset('ami','headset-single', revision='1.8.0')
内容的提问来源于stack exchange,提问作者Patrick
相关产品推荐
相关产品推荐

