记录一下调试MambaVisionMixer的错误
这个错误通常发生在使用 PyTorch 的 .view()
方法时,当你试图将一个张量重塑为一个新的形状,但这个新的形状与原始张量的内存布局不兼容。具体来说,如果张量的某些维度不是连续的(即,它们在内存中的排列不是连续的),那么你不能直接通过 .view()
方法来改变其形状,因为 .view()
要求新的形状在内存中是可行的。
在调用 .view()
之前,尝试使用 .contiguous()
方法。这个方法会返回一个内存连续的新张量,这样你就可以安全地使用 .view()
了。
【无标题】
最新推荐文章于 2024-09-11 18:01:31 发布