1. 错误处理
首先出现了什么错误,如图所示,我解决的办法是:将报错的所有AT_CHECK全部改为TORCH_CHECK。
解决方案:AT_CHECK错误
当你修改大概40多处的问题就可以安装上那个可选的第五步东东了。
在提一嘴,我在terminal中无法直接安装那个,我是在github上下载解压后,运行
python setup.py develop
才运行成功的。
之后还需要在网上下载几个pth数据,最后运行demo成功。
最后只需要在文件夹中找demo运行的文件即可。
运行成功,今天进行训练,顺风顺水训练五个小时,损失函数一点都不变,心态炸裂。经过查看github代码的解决方案,需要在train.yaml文件的一行加上点东西
加的东西是76行。结果如下图: