ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 48])

训练数据在加载的时候遇到batch=1的情况,可以设定DataLoader中的drop_last=True解决

你可能感兴趣的:(错误合集,pytorch)