%20(2).svg)
项目简介
本项目聚焦于复现论文“LEARN Codes: Inventing Low-latency Codes via Recurrent Neural Networks”，并在原有模型基础上进行改进，增加了attention机制。
项目的主要特性和功能
- 实现论文模型及训练流程的复现。
- 对原模型改进，加入attention机制，提升模型性能。
- 可通过命令行参数灵活配置训练参数，具体参数配置参考
get_args.py文件。
安装使用步骤
假设用户已下载本项目的源码文件，可按以下步骤安装和使用：
1. 复制项目并进入项目目录：
bash
2. 设置环境（以下两种方式选择一种即可）：
- 使用environment.yaml文件创建环境：
bash
conda env create -f environment.yaml
- 手动创建虚拟环境并安装依赖：
bash
conda create -n pytorch python=3.8.3 # 创建新的虚拟环境
source activate pytorch # 激活新建的虚拟环境
pip install -r requirements.txt # 安装对应依赖
3. 执行主程序：
bash
python main.py -num_epoch 120
main.py运行时还可更改其他参数，具体详见get_args.py。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】