1、问题描述
在测试单个实验数据时报错Expected more than 1 value per channel when training, got input size torch.Size([]),猜测应该是BN层出现了问题。并且通过model.eval()后发现没有解决该问题。
2、解决办法
通过检查网络结构后,发现和BatchNorm1d中track_running_stats设置有很大关系,设置track_running_stats=True解决问题。
在测试单个实验数据时报错Expected more than 1 value per channel when training, got input size torch.Size([]),猜测应该是BN层出现了问题。并且通过model.eval()后发现没有解决该问题。
通过检查网络结构后,发现和BatchNorm1d中track_running_stats设置有很大关系,设置track_running_stats=True解决问题。