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