LAS_Mandarin_PyTorch 项目使用教程

LAS_Mandarin_PyTorch 项目使用教程

LAS_Mandarin_PyTorch Listen, attend and spell Model and a Chinese Mandarin Pretrained model (中文-普通话 ASR模型) 项目地址: https://gitcode.com/gh_mirrors/la/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: 定义训练过程中的批量大小、学习率、训练轮数等参数。

通过修改配置文件，可以灵活调整模型的训练和测试参数。

LAS_Mandarin_PyTorch Listen, attend and spell Model and a Chinese Mandarin Pretrained model (中文-普通话 ASR模型) 项目地址: https://gitcode.com/gh_mirrors/la/LAS_Mandarin_PyTorch