RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2

作者：你好赵伟 | 2024-02-27 03:18:32

踩

runtimeerror: sizes of tensors must match except in dimension 1.

在训练语义分割模型时，出现以下错误：

在模型中有以下操作：torch.cat(x, y)

当x, y的尺寸不一致时，就会出现以上错误。

进行单步调试，发现我的错误是因为图像山采样和下采样后，由于取整的方式不一样，导致的x、y的尺寸不一致。

方法1：在图像输入时，宽和高最好是32的整数倍，就不会出现取整的问题。

方法2：把x、y resize到相同的尺寸。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/149616