🚀 wav2vec2-large-xls-r-300m-hi-CV7
本模型是基于 facebook/wav2vec2-xls-r-300m 在 MOZILLA-FOUNDATION/COMMON_VOICE_7_0 - HI 数据集上微调得到的。它在语音自动识别任务中表现出色,能够有效处理相关语音数据,为语音识别领域提供了可靠的解决方案。
✨ 主要特性
- 支持印地语(hi)的自动语音识别。
- 基于
wav2vec2-large-xls-r-300m
模型微调,在特定数据集上有较好表现。
- 可用于
Automatic Speech Recognition
任务,在多个评估指标上有明确的评估结果。
📦 安装指南
文档未提供安装步骤,此部分跳过。
💻 使用示例
文档未提供代码示例,此部分跳过。
📚 详细文档
评估指标
该模型在评估集上取得了以下结果:
- 损失值(Loss):0.6588
- 词错误率(Wer):0.2987
评估命令
- 在
mozilla-foundation/common_voice_8_0
测试集上进行评估
python eval.py --model_id DrishtiSharma/wav2vec2-large-xls-r-300m-hi-CV7 --dataset mozilla-foundation/common_voice_7_0 --config hi --split test --log_outputs
- 在
speech-recognition-community-v2/dev_data
上进行评估
暂无可用评估命令(NA)
训练超参数
训练过程中使用了以下超参数:
- 训练批次大小(train_batch_size):16
- 评估批次大小(eval_batch_size):8
- 随机种子(seed):42
- 梯度累积步数(gradient_accumulation_steps):2
- 总训练批次大小(total_train_batch_size):32
- 优化器(optimizer):Adam(β1 = 0.9,β2 = 0.999,ε = 1e-08)
- 学习率调度器类型(lr_scheduler_type):线性
- 学习率调度器热身步数(lr_scheduler_warmup_steps):2000
- 训练轮数(num_epochs):60
- 混合精度训练(mixed_precision_training):Native AMP
训练结果
训练损失 |
轮数 |
步数 |
验证损失 |
词错误率(Wer) |
12.809 |
1.36 |
200 |
6.2066 |
1.0 |
4.3402 |
2.72 |
400 |
3.5184 |
1.0 |
3.4365 |
4.08 |
600 |
3.2779 |
1.0 |
1.8643 |
5.44 |
800 |
0.9875 |
0.6270 |
0.7504 |
6.8 |
1000 |
0.6382 |
0.4666 |
0.5328 |
8.16 |
1200 |
0.6075 |
0.4505 |
0.4364 |
9.52 |
1400 |
0.5785 |
0.4215 |
0.3777 |
10.88 |
1600 |
0.6279 |
0.4227 |
0.3374 |
12.24 |
1800 |
0.6536 |
0.4192 |
0.3236 |
13.6 |
2000 |
0.5911 |
0.4047 |
0.2877 |
14.96 |
2200 |
0.5955 |
0.4097 |
0.2643 |
16.33 |
2400 |
0.5923 |
0.3744 |
0.2421 |
17.68 |
2600 |
0.6307 |
0.3814 |
0.2218 |
19.05 |
2800 |
0.6036 |
0.3764 |
0.2046 |
20.41 |
3000 |
0.6286 |
0.3797 |
0.191 |
21.77 |
3200 |
0.6517 |
0.3889 |
0.1856 |
23.13 |
3400 |
0.6193 |
0.3661 |
0.1721 |
24.49 |
3600 |
0.7034 |
0.3727 |
0.1656 |
25.85 |
3800 |
0.6293 |
0.3591 |
0.1532 |
27.21 |
4000 |
0.6075 |
0.3611 |
0.1507 |
28.57 |
4200 |
0.6313 |
0.3565 |
0.1381 |
29.93 |
4400 |
0.6564 |
0.3578 |
0.1359 |
31.29 |
4600 |
0.6724 |
0.3543 |
0.1248 |
32.65 |
4800 |
0.6789 |
0.3512 |
0.1198 |
34.01 |
5000 |
0.6442 |
0.3539 |
0.1125 |
35.37 |
5200 |
0.6676 |
0.3419 |
0.1036 |
36.73 |
5400 |
0.7017 |
0.3435 |
0.0982 |
38.09 |
5600 |
0.6828 |
0.3319 |
0.0971 |
39.45 |
5800 |
0.6112 |
0.3351 |
0.0968 |
40.81 |
6000 |
0.6424 |
0.3252 |
0.0893 |
42.18 |
6200 |
0.6707 |
0.3304 |
0.0878 |
43.54 |
6400 |
0.6432 |
0.3236 |
0.0827 |
44.89 |
6600 |
0.6696 |
0.3240 |
0.0788 |
46.26 |
6800 |
0.6564 |
0.3180 |
0.0753 |
47.62 |
7000 |
0.6574 |
0.3130 |
0.0674 |
48.98 |
7200 |
0.6698 |
0.3175 |
0.0676 |
50.34 |
7400 |
0.6441 |
0.3142 |
0.0626 |
51.7 |
7600 |
0.6642 |
0.3121 |
0.0617 |
53.06 |
7800 |
0.6615 |
0.3117 |
0.0599 |
54.42 |
8000 |
0.6634 |
0.3059 |
0.0538 |
55.78 |
8200 |
0.6464 |
0.3033 |
0.0571 |
57.14 |
8400 |
0.6503 |
0.3018 |
0.0491 |
58.5 |
8600 |
0.6625 |
0.3025 |
0.0511 |
59.86 |
8800 |
0.6588 |
0.2987 |
框架版本
- Transformers:4.16.2
- Pytorch:1.10.0+cu111
- Datasets:1.18.3
- Tokenizers:0.11.0
🔧 技术细节
文档未提供技术实现细节,此部分跳过。
📄 许可证
本模型采用 Apache-2.0 许可证。
📋 模型信息
属性 |
详情 |
模型类型 |
用于印地语自动语音识别的微调模型 |
训练数据 |
mozilla-foundation/common_voice_7_0 |