问题
环境
python 3.8
Tensorflow 2.3
使用Tensorflow2版本报错Found unexpected keys that do not correspond to any Model output: dict_keys([‘model_output’]). Expected: [‘dense’]
解决
因为Tensorflow2的model使用的最后一层需要命名比如model_ouput,则对应并且在model.fit_generator的生成器函数中返回值要定义相应的字典格式,比如{‘model_ouput’:ouput_value}。就是这两个地方要一样
在Model中代码如下
在生成器函数中