训练数据在加载的时候遇到batch=1的情况,可以设定DataLoader中的drop_last=True解决
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 48])
最新推荐文章于 2024-07-25 13:59:35 发布
训练数据在加载的时候遇到batch=1的情况,可以设定DataLoader中的drop_last=True解决