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

1. 背景介绍
      在训练编码解码网络模型时,出现以下错误:

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

2. 错误产生原因
      在模型中有以下操作:torch.cat(x, y)

当x, y的尺寸不一致时,就会出现以上错误。原因是上下采样过程中四舍五入造成的tensor相加尺寸问题,

3. 解决方法
      方法1:在图像输入时,宽和高最好是32的整数倍,就不会出现取整的问题。

      方法2:把x、y 调整到相同的尺寸。
 

你可能感兴趣的:(深度学习,深度学习,人工智能)