合并已保存Keras模型时遇TypeError问题求助
问题解决:多输出函数式模型fit时的结构不匹配错误
错误原因
你遇到的TypeError是因为模型输入输出的结构定义错误:
model_1.inputs本身是列表类型(即使模型只有一个输入),你用[model_1.inputs]会将其嵌套为「列表的列表」,导致模型输入结构与实际传入的单个张量X不匹配。- 同理
model_1.outputs也是列表,嵌套后输出结构也会出现层级错误,最终触发输入输出结构不兼容的报错。
修正代码
1. 正确构建多输出模型
替换原模型构建代码为以下两种方式之一:
方式一:直接使用张量属性
# 直接获取单个输入/输出张量 input_tensor = model_1.input mid_output = model_1.output final_output = model_2(mid_output) # 定义多输出模型 model = Model(inputs=input_tensor, outputs=[mid_output, final_output])
方式二:从列表中取单个元素
# 从inputs/outputs列表中提取单个张量 model = Model( inputs=model_1.inputs[0], outputs=[model_1.outputs[0], model_2(model_1.outputs[0])] )
2. 修正fit调用
fit时无需将X包裹在列表中,直接传入单个张量:
model.fit(x=X, y=[X, Y], epochs=10)
验证说明
修改后模型的输入为单个(28,28,1)张量,输出为两个张量(分别对应中间层和最终分类层),与你传入的y=[X,Y]结构完全匹配,编译和训练都能正常运行,同时保留了对中间层输出计算损失的需求。
内容的提问来源于stack exchange,提问作者AmanKP
相关产品推荐
相关产品推荐

