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 调整到相同的尺寸。