RuntimeError: 1only batches of spatial targets supported (3D tensors) but got targets of size: : [1]
这个原因主要是因为,有个tensor 是三维的,但实际上代码需要是二维的tensor (我的代码遇到的问题是这样,网上众说纷纭,但没看到我需要的答案)
首先找到代码报错那行,然后,看看哪个是三维的,需要转换成二维
tensor: [[[[0.9033, 1.0250, 0.8474, 0.9353, 1.0470]]]] # 这种是错误的
它的size是(1,1,5),直接view, 比如是a这个tensor,直接a.view(1,5)或者a.reshape(1,5)