Fix make_image_classifier.

PiperOrigin-RevId: 296228856
ChanchalKumarMaji · Feb 24, 2020 · 571bb28 · 571bb28
1 parent 7f7d411
commit 571bb28
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorflow_hub/tools/make_image_classifier/make_image_classifier_lib.py b/tensorflow_hub/tools/make_image_classifier/make_image_classifier_lib.py
@@ -180,14 +180,13 @@ def build_model(module_layer, hparams, image_size, num_classes):
   """
   # TODO(b/139467904): Expose the hyperparameters below as flags.
   model = tf.keras.Sequential([
-      module_layer,
+      tf.keras.Input(shape=(image_size[0], image_size[1], 3)), module_layer,
       tf.keras.layers.Dropout(rate=hparams.dropout_rate),
       tf.keras.layers.Dense(
           num_classes,
           activation="softmax",
           kernel_regularizer=tf.keras.regularizers.l2(0.0001))
   ])
-  model.build((None, image_size[0], image_size[1], 3))
   print(model.summary())
   return model