TypeError: cat() got an unexpected keyword argument ‘axis‘

这个问题是跑b站作者bubbliiiing写的YOLOv4（pytorch版本）代码中出现的问题，在yolo4.py中有一个特征层之间的连接过程
P4 = torch.cat( [P4,P5_upsample], axis=1)，
作者给出的常见问题汇总给出的建议是将pytorch版本更新到1.2.0以上，但是我的显卡是老显卡，cuda版本9.0，查看了版本对应表，最高也只能安装1.0.1版本，但是对于旧版本应该也是可以用的，只要将axis去掉或者改为dim就可以了，具体为
P4 = torch.cat( [P4,P5_upsample], 1)，
P4 = torch.cat( [P4,P5_upsample], dim=1)。