这个问题是在dataloader遇到的,网上一般是数据的维度不同,但我这里其实是最后一份数据和之前的数据比数量不足,所以凑不够一个batch,所以我的做法是加入drop_last=True意思是如果数据不够就舍弃最后一个batch。
pytorch报错: invalid argument 0: Sizes of tensors must match except in dimension 0.
猜你喜欢
转载自blog.csdn.net/qq_32146369/article/details/103321203
今日推荐
周排行