AttributeError: 'Node' object has no attribute 'output_masks'

后端 未结 4 1969
情歌与酒
情歌与酒 2021-01-04 08:02

I use Keras pretrained model VGG16. The problem is that after configuring tensorflow to use the GPU I get an error that I didn\'t have before when using the CPU.

The

4条回答
  •  臣服心动
    2021-01-04 08:35

    I had a similar issue, but with different architecture. As people suggested, it's important not to mix keras with tensorflow.keras, so try swapping code like:

    from keras.preprocessing import image
    from keras.models import Model
    from keras.layers import Dense, GlobalAveragePooling2D
    from keras import backend as K
    

    to:

    from tensorflow.keras.preprocessing import image 
    from tensorflow.keras.models import Model 
    from tensorflow.keras.layers import Dense, GlobalAveragePooling2D 
    from tensorflow.keras import backend as K
    

    Also make sure, you don't use keras.something inside your code (not only imports) as well, hope it helps : ) Also, I used Keras 2.2.4 with tensorflow 1.10.0

提交回复
热议问题