view size is not compatible with input tensor‘s size and stride

最近在跑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(),终于跑通了。激动!!!

  • 17
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值