ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,1,1]
解决办法:
在DataLoader类里面加上:drop_last=True
已解决。
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,1,1]
解决办法:
在DataLoader类里面加上:drop_last=True
已解决。