问题描述:现有模型类别4种,增加新的类型后报错。
Node: 'categorical_crossentropy/softmax_cross_entropy_with_logits'
logits and labels must be broadcastable: logits_size=[45,4] labels_size=[45,5]
基础环境:
Python版本是3.9.13;使用的神经网络是CNN;所用到的搭建网络层的库是Keras。报错位置为下图方法。
模型类别数量增加后开始报错,各种寻找原因最后发现因为h5文件中记录的分类数量和最新的模型数量不匹配。备份原有*.h5文件,重新生成模型。