Dual-Path-RNN-Pytorch 项目教程
Dual-Path-RNN-Pytorch项目地址:https://gitcode.com/gh_mirrors/du/Dual-Path-RNN-Pytorch
1. 项目的目录结构及介绍
Dual-Path-RNN-Pytorch/
├── config/
│ ├── Conv_Tasnet/
│ └── Dual_RNN/
├── data_loader/
├── log/
│ └── Conv_Tasnet/
├── logger/
├── model/
├── trainer/
├── utils/
├── .gitignore
├── LICENSE
├── README.md
├── create_scp.py
├── dualrnn_test.py
├── dualrnn_test_wav.py
├── requirements.txt
├── test_tasnet.py
├── test_tasnet_wav.py
├── train_Tasnet.py
├── train_rnn.py
└── utils.py
目录结构介绍
config/
: 包含项目的配置文件,分为Conv_Tasnet
和Dual_RNN
两个子目录。data_loader/
: 数据加载相关的文件。log/
: 日志文件存储目录,目前包含Conv_Tasnet
的日志。logger/
: 日志记录相关的文件。model/
: 模型定义文件。trainer/
: 训练器相关的文件。utils/
: 工具函数文件。.gitignore
: Git 忽略文件配置。LICENSE
: 项目许可证。README.md
: 项目说明文档。create_scp.py
: 生成 scp 文件的脚本。dualrnn_test.py
: 多音频测试脚本。dualrnn_test_wav.py
: 单音频测试脚本。requirements.txt
: 项目依赖文件。test_tasnet.py
: Conv-TasNet 测试脚本。test_tasnet_wav.py
: Conv-TasNet 单音频测试脚本。train_Tasnet.py
: Conv-TasNet 训练脚本。train_rnn.py
: Dual-Path-RNN 训练脚本。utils.py
: 通用工具函数文件。
2. 项目的启动文件介绍
训练脚本
train_Tasnet.py
: 用于训练 Conv-TasNet 模型。train_rnn.py
: 用于训练 Dual-Path-RNN 模型。
测试脚本
dualrnn_test.py
: 多音频测试脚本。dualrnn_test_wav.py
: 单音频测试脚本。test_tasnet.py
: Conv-TasNet 测试脚本。test_tasnet_wav.py
: Conv-TasNet 单音频测试脚本。
3. 项目的配置文件介绍
配置文件目录
config/Conv_Tasnet/train.yml
: Conv-TasNet 模型的训练配置文件。config/Dual_RNN/train.yml
: Dual-Path-RNN 模型的训练配置文件。
配置文件示例
# config/Conv_Tasnet/train.yml
# 配置文件示例
batch_size: 16
learning_rate: 0.001
epochs: 100
# 其他配置项...
# config/Dual_RNN/train.yml
# 配置文件示例
batch_size: 32
learning_rate: 0.0005
epochs: 200
# 其他配置项...
使用方法
-
生成 scp 文件:
python create_scp.py
-
修改训练和模型参数:
cd config/Dual_RNN vim train.yml
-
启动训练:
python train_rnn.py --opt config/Dual_RNN/train.yml
通过以上步骤,您可以配置和启动 Dual-Path-RNN-Pytorch 项目的训练和测试。
Dual-Path-RNN-Pytorch项目地址:https://gitcode.com/gh_mirrors/du/Dual-Path-RNN-Pytorch