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

pytorch训练时报错：
报错内容：ValueError:Expected more than 1 value per channel when training, got input size torch.Size([1,1024])

报错原因：网络中含有bn或dropout，训练模式中bn和dropout的行为方式和预测模式是不一样的。

解决方法：
在训练部分的代码前面加上：model.train()
在测试部分的代码前面加上：model.eval()