“基于预训练的自然语言生成以进行文本汇总”的实现
论文: :
版本号
python 2.7
PyTorch:1.0.1.post2
准备包/数据集
运行: pip install -r requirements.txt以安装所需的软件包
从以下位置下载块CNN / DailyMail数据: :
运行: python news_data_reader.py创建将在我的数据加载器中使用的泡菜文件
运行模型
对我来说,该模型对我的GPU来说太大了,因此我使用了以下较小的参数进行调试。 CUDA_VISIBLE_DEVICES=3 python main.py --cuda --batch_size=2 --hop 4 --hidden_dim 100
审阅者须知:
尽管我实现了核心部分(使用BERT进行两步汇总生成),但是我没有足够的时间来实现RL部分。
第二个解码器过程非常耗时
2022-08-06 17:52:07
26KB
Python
1