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

解决方法:
训练时的batchsize由1改为4

猜你喜欢

转载自blog.csdn.net/weixin_44025103/article/details/129404575