search

titu1994 / keras-efficientnets

Keras Implementation of EfficientNets
MIT License
188 stars 63 forks source link

No attribute 'swish' error #9

Closed jannaescur closed 5 years ago

jannaescur commented 5 years ago

I get the following error: AttributeError: module 'tensorflow.python.ops.nn' has no attribute 'swish'

Tensorflow version: 1.4.0 (GPU, installed before your library) Keras version: 2.2.4

I have imported the model: from keras_efficientnets import EfficientNetB4 This should import custom_objects too, right?

Any idea of what can be happening? Thanks

titu1994 commented 5 years ago

Update your tensorflow version. TF.nn.swish was not added in TF 1.4