ValueError:expected sequence of length 10 at dim 1 (got 1)
在数据预处理创建mini batch时,因为以下代码导致出错:ValueError:expected sequence of length 10 at dim 1 (got 1) inout_seq.append((train_seq, train_label)) return torch.FloatTensor(inout_seq)原因是train_seq和 train_label 长度一不一样,一个有10个元素,另一个只有一个。修改好的办法是, 在划分trainset, test
原创
2020-10-29 19:56:24 ·
31871 阅读 ·
5 评论