python报错系列(7)--ValueError: Expected more than 1 value perchannel when training, got input size ....

文章目录

• 前言
• 1、ValueError: Expected more than 1 value per channel when training, got input size ([1, 256, 1, 1])
• 2、解决方式
• 总结

前言

1、ValueError: Expected more than 1 value per channel when training, got input size ([1, 256, 1, 1])

大致意思是，由模型中BN层属性决定，训练完train样本后，生成的模型model要用来测试样本。在model(test)之前，需要加上()，否则的话，有输入数据，即使不训练，它也会改变权值。

2、解决方式

在测试之前加上()

备用思路1参考：
模型中用了batchnomolization，训练中用batch训练的时候，应该是有单数，比如dataset的总样本数为17,你的batch_size为8,就会报这样的错误。
解决方式：
从dataset中删掉一个sample。
备用思路2参考：
看上去是因为 BN层里，遇到了batch=1的情况
解决方式：
设置 batch_size>1, 且 drop_last=True

DataLoader(train_set, batch_size=args.train_batch_size,
          num_workers=args.num_workers, shuffle=(train_sampler is None), 
          drop_last=True, sampler = train_sampler)

---

总结

提示：这里对文章进行总结：
例如：以上就是今天要讲的内容，本文仅仅简单介绍了pandas的使用，而pandas提供了大量能使我们快速便捷地处理数据的函数和方法。