最近在跑py-Vital的代码
读论文这么久了,第一次跑代码.啥也不懂,
上来一开始跑了tracking,一直报错。觉得代码应该没有问题啊。以为是没有跑预训练的缘故,于是,又去下载数据集。当然下载很慢,直接把别人服务器上的代码cp过来,或者直接改了地址,引用了别人文件夹下的数据集,亲测好用,嘿嘿。
跑了两天预训练,终于跑完了,又去跑tracking,又报错,还是同样的错误。
RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.
参考别人的解决方法:原因:用多卡训练的时候tensor不连续,即tensor分布在不同的内存或显存中。
解决方法:对tensor进行操作时先调用contiguous()。如tensor.contiguous().view()
于是打开文档,ctrl+f ,查找view,在前面加上contiguous(),终于跑通了。激动!!!