情况一：
作为输入的图像数据的维度不符合模型的要求，常用的输入大小有224*224，笔者暂时没找到精确知晓模型所需输入图像数据维度大小的方法，只能凭借经验尝试。若有大牛知晓，烦请告知。此外，还有一种可能是数据集里面图像维度不一，因此最好进入模型前对图像数据维度进行处理。
情况二：
batch_size可能是导致该问题的因素之一，简单来说就是数据集总数不能整除batch_size，最后导致维度不匹配。因此，可以尝试将batch_size设置为1。
综上所述，当出现该问题时，先将batch_size设置为1，若问题没有得到解决，尝试情况一的解决方法。
