LAS_Mandarin_PyTorch 项目使用教程
1. 项目的目录结构及介绍
LAS_Mandarin_PyTorch/
├── ckpt/
│ └── aishell_asr_example_sd0/
├── config/
├── core/
├── dataset/
├── misc/
├── LICENSE
├── README.md
├── README_zh-CN.md
├── aishell_vocab.txt
├── eval.py
├── generate_vocab_file.py
├── generate_vocab_file_aishell.py
├── main.py
├── requirements.txt
├── test_asr.py
├── train_asr.py
└── train_lm.py
目录结构介绍
- ckpt/: 存放预训练模型和训练过程中的检查点文件。
aishell_asr_example_sd0/
: 示例的ASR模型检查点文件。
- config/: 存放项目的配置文件。
- core/: 存放项目核心代码,包括模型定义、数据处理等。
- dataset/: 存放数据集处理相关的代码和脚本。
- misc/: 存放其他辅助工具和脚本。
- LICENSE: 项目的开源许可证文件。
- README.md: 项目的英文说明文档。
- README_zh-CN.md: 项目的中文说明文档。
- aishell_vocab.txt: 中文词汇表文件。
- eval.py: 用于评估模型的脚本。
- generate_vocab_file.py: 生成词汇表文件的脚本。
- generate_vocab_file_aishell.py: 生成AISHELL数据集词汇表文件的脚本。
- main.py: 项目的启动文件。
- requirements.txt: 项目依赖的Python包列表。
- test_asr.py: 用于测试ASR模型的脚本。
- train_asr.py: 用于训练ASR模型的脚本。
- train_lm.py: 用于训练语言模型的脚本。
2. 项目的启动文件介绍
main.py
main.py
是项目的启动文件,负责初始化模型、加载配置、启动训练或测试等任务。以下是该文件的主要功能:
- 初始化模型: 根据配置文件初始化ASR模型。
- 加载数据: 加载训练或测试数据集。
- 训练模型: 启动模型的训练过程。
- 测试模型: 对训练好的模型进行测试。
使用方法
python main.py --config config/your_config_file.yaml
3. 项目的配置文件介绍
config/
config/
目录下存放了项目的配置文件,通常以 .yaml
或 .json
格式存储。配置文件定义了模型的超参数、数据路径、训练参数等。
配置文件示例
model:
name: "LAS"
hidden_size: 512
num_layers: 3
data:
train_path: "dataset/train.csv"
test_path: "dataset/test.csv"
vocab_path: "aishell_vocab.txt"
training:
batch_size: 32
learning_rate: 0.001
epochs: 10
配置文件说明
- model: 定义模型的名称、隐藏层大小、层数等。
- data: 定义训练和测试数据的路径,以及词汇表文件的路径。
- training: 定义训练过程中的批量大小、学习率、训练轮数等参数。
通过修改配置文件,可以灵活调整模型的训练和测试参数。