使用模型时，batchsize为1，模型报错

使用训练后模型，batchsize为1，模型报错

本文分享一个很新手向的问题。由于自己刚开始接触Pytorch，有些问题是第一次遇到，因此分享一下这次经验，作为一个笔记，同时也希望能帮助到有需要的伙伴。

在训练完自己的模型之后，然后就需要在算法中调用此模型。在某些场景下，需要求解的是一个测试例，也就是相当于batchsize的大小为1，此时将这样一个单独的数据喂给模型，会发现报如下错：

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

查阅相关内容得知，因为自己的模型中涉及到BatchNorm的操作，进行BatchNorm的操作需要一个以上的数据，而此时喂给模型只有一个数据，因此会报此种错误。

解决办法

但是在某些场景下，需要使用模型时，只需要一个输入数据就行，那该如何处理呢。其实很简单，在使用前加个model.eval()就可以解决此问题。

因为自己刚开始接触pytorch，没有太在意 model.train() 和 model.eval() 的作用，关于这两个语句的作用，有很多优质的博客进行了讲解，有兴趣的伙伴可以进行查阅。在本场景下，mode.eval()之后，可以理解成进行评估模式，batchnorm层会被关闭，此时就可以顺利地调用此模型了。