SCI211985的博客

原创 代码报错解决

超出了数据集的长度。遇到的错误，"stack expects each tensor to be equal size"，很可能是因为当批量大小超过数据集末尾时，会出现不同大小的张量。你的批量大小为12，如果在数据集末尾剩下的样本少于12个，你会得到形状为 [12, 6000] 和 [n, 6000] 的张量，其中 'n' 小于12。解决办法：在def __getitem__(self, index): 下添加以下代码。方法中添加的代码用于处理情况，其中。