有些冷门算法真的不要随便pip,有可能下载到史前版本…最好还是找源代码 汗
今天要用到SSIM损失函数,从网上简单看了一下原理就想测试一下,偷了一下懒就直接在命令行输入pip install pytorch_ssim了,结果报了一堆错误(汗),正如这篇博文所说博文在此,传入Conv2D中的padding是浮点数,然而这样做完之后还会报错,差不多是说img和window不在一个运行设备上,我这边是window张量没有在gpu上,结果我用了.cuda()之后还是不管用,然后我就去github上看了一下,才发现这个pip包的版本太落后了。于是我重新下载覆盖之后,成功运行了!感谢!
点我github源码
后话:如果你想使用github上给出的案例,请你仿照知乎上这位给出的改进版本,源代码会报错。点这里