Closed SamuelMarks closed 5 years ago
Got the same issue with the following code (executed on a kaggle notebook):
!pip install -U --pre efficientnet
from efficientnet.tfkeras import EfficientNetB0
import tensorflow as tf
inputs = tf.keras.layers.Input((None, None, 3))
base_model = EfficientNetB0(weights='imagenet',include_top=False, classes=2, input_tensor=inputs)
outputs = base_model.output
model = tf.keras.Model(inputs=[inputs], outputs=[outputs])
model.summary()