【论文复现】QAFactEval: 《QAFactEval: Improved QA-Based Factual Consistency Evaluation for Summarization》

本文介绍了如何在2021年NAACL会议上提出的QAFactEval代码在3.0.2版本的transformers库环境下运行的困难,特别是处理facebook/bart-large模型加载问题,以及作者提供的解决方法,包括创建虚拟环境、安装依赖库、修改源代码等。

论文《QAFactEval: Improved QA-Based Factual Consistency Evaluation for Summarization》(NAACL 2022)代码https://github.com/salesforce/QAFactEval

本人的服务器:学校实验室服务器,不支持正常访问huggingface

因为本论文的时间是2021年,当时使用的transformers库为3.0.2版本的,并且按照代码的框架逻辑,不能高于3.1.0版本,否则要大改特改。

transformers库最恶心的地方就在于,更新换代的速度非常之快,并且新版本的修改可谓是大刀阔斧的修改,很多旧有的方法不再支持。

对于本代码而言,困难之处在于:

  • 在封装好的库qafacteval中写死了必须远程加载facebook/bart-large模型和tokenizer,并且加载facebook/bart-large的那几行代码无法找到。并且是基于3.0.2版本的加载,加载的逻辑已经不同于现在的transformers库。

总结复现流程如下:

  1. 在服务器上conda创建虚拟环境qafacteval。(亲测,python版本不能超过3.8,作者的readme里没说qwq)

    conda create -n questeval python=3.8
    
  2. git clone下载项目代码于本地,用pycharm打开并远程连接到服务器的该环境中。

  3. 服务器上进入该项目目录,安裝本项目需要的库。

    pip install -e .
    
  4. 还需要再安装库gdown,不然待会儿没法下载模型。(这个作者没在setup.py中实现,也没另外说qwq)

    pip install gdown
    
  5. 执行download_models.sh,下载本框架所依赖的所有子模型。(其实还依赖了bert-large作为生成模型,被隐藏在库里面了,这是最搞人的地方)

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值