admin 管理员组

文章数量: 887038

‘Keyword argument not understood:‘, ‘keepdims‘ 解决方案

报错原因

由于tensorflow和keras的更新,导致tensorflow在

GlobalMaxPooling2D

这个方法中需要使用 keepdims这个参数

由于我是在使用迁移学习fine-tuning这个场景下出的这个问题,而我确实是修改了 GlobalMaxPooling2D,所以只需要改成

output = keras.layers.GlobalMaxPooling2D(keepdims=True)(output)

本文标签: Keyword argument not understood keepdims 解决方案