当前位置: 代码迷 >> 综合 >> pytorch RuntimeError: input image (T: 9 H: 8 W: 2) smaller than kernel size (kT: 4 kH: 4 kW: 4)
  详细解决方案

pytorch RuntimeError: input image (T: 9 H: 8 W: 2) smaller than kernel size (kT: 4 kH: 4 kW: 4)

热度:28   发布时间:2023-12-15 17:09:44.0

报错:RuntimeError: input image (T: 9 H: 8 W: 2) smaller than kernel size (kT: 4 kH: 4 kW: 4)

原因:可能不是字面那样是输入图片尺寸小了,而是维度不匹配。

比如需要的输出是 (channel, height, width, frames)你输入的是 (channel, frames, height, width)

那就对不上了。

解决:改一下dataloader中的组装data的代码,将维度顺序调整一下。可以用tensor.permute()。

  相关解决方案