YancyBill的博客

原创 测试时模型时出现：Expected more than 1 value per channel when training, got input size torch.Size([])问题

在测试单个实验数据时报错Expected more than 1 value per channel when training, got input size torch.Size([])，猜测应该是BN层出现了问题。并且通过model.eval()后发现没有解决该问题。