ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])
改batchsize的大小 ,如果写1的话没成,可以做参考
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512,
最新推荐文章于 2024-04-26 15:47:14 发布