当前位置: 代码迷 >> 综合 >> ‘Keyword argument not understood:‘, ‘keepdims‘ 解决方案
  详细解决方案

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

热度:8   发布时间:2023-12-29 00:05:43.0

报错原因

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

GlobalMaxPooling2D

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

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

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

  相关解决方案