报错原因
由于tensorflow和keras的更新,导致tensorflow在
GlobalMaxPooling2D
这个方法中需要使用 keepdims这个参数
由于我是在使用迁移学习fine-tuning这个场景下出的这个问题,而我确实是修改了 GlobalMaxPooling2D,所以只需要改成
output = keras.layers.GlobalMaxPooling2D(keepdims=True)(output)
由于tensorflow和keras的更新,导致tensorflow在
GlobalMaxPooling2D
这个方法中需要使用 keepdims这个参数
由于我是在使用迁移学习fine-tuning这个场景下出的这个问题,而我确实是修改了 GlobalMaxPooling2D,所以只需要改成
output = keras.layers.GlobalMaxPooling2D(keepdims=True)(output)