今天早上跑通了 PF-AFN！

项目地址：PF-AFN

上午de了两个小时的bug，终于把模型跑通了！

官方的result：

我机器上的result：

哈哈哈哈哈！

记录下自己从github上下载模型后开始到能运行模型的整个debug过程。

Debug之路

先是搭建模型运行环境，主要是GPU-PyTorch和CUDA。安装GPU-PyTorch花了很长时间，后来在某宝上花了30RMB请人帮忙装了一个😂。

从CUDA官网安装CUDA速度实在太慢，后来就直接在网上搜了网友已经下载好的CUDA安装包，于是CUDA的安装问题就顺利解决了，感谢热心网友！有必要提醒一句的是，CUDA的版本一定要和自己电脑的GPU版本适配。

这个模型需要用到cupy，下载与自己电脑上CUDA版本适配的cupy版本即可。

然后就是跑模型了，我是在PyCharm中打开这个项目的，尝试过在终端运行python test.py 结果不行，老是GPU空间不足的异常，后来直接复制项目里自带的test.sh中的命令。然后报了有未预料到的XX线程存在的异常，后来了解到是开了多线程，于是通过改代码将多线程关掉。

再次跑测试后，还是会报GPU空间不足的异常。在查资料后，将test.py中的

gen_outputs = gen_model(gen_inputs)

改为

with torch.no_grad():
    gen_outputs = gen_model(gen_inputs)

模型终于成功跑通！