number of dims don't match in permute

[RobertXiaobinW]

楼主楼主
报错如下：
90 x = self.cnn(x) # [B,512,W/16,1]
91 x = torch.squeeze(x, 3) # [B,512,W]
---> 92 x = x.permute([0, 2, 1]) # [B,W,512]
93 x, h1 = self.rnn1(x)
94 x, h2 = self.rnn2(x, h1)

RuntimeError: number of dims don't match in permute

是因为我前面CTPN程序里的图片裁得太细了吗？换了张大点的图片可以呢

图片尺寸：(2581, 276, 3)
(2580, 283, 3)
(2545, 257, 3)
(2058, 321, 3)

[yizt]

@RobertXiaobinW 您好，可以在91行打印出x的维度，看看是什么问题