Internal change

PiperOrigin-RevId: 398612564

Internal change
PiperOrigin-RevId: 398612564
24b2ccd4 · Rebecca Chen · A. Unique TensorFlower · e5fbe328 · 24b2ccd4
Commit 24b2ccd4 authored Sep 23, 2021 by Rebecca Chen Committed by A. Unique TensorFlower Sep 23, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/image_classification/efficientnet/efficientnet_model.py ...n/image_classification/efficientnet/efficientnet_model.py +1 -1

No files found.
--- a/official/vision/image_classification/efficientnet/efficientnet_model.py
+++ b/official/vision/image_classification/efficientnet/efficientnet_model.py
@@ -327,7 +327,7 @@ def mb_conv_block(inputs: tf.Tensor,
  return x


-def efficientnet(image_input: tf.keras.layers.Input, config: ModelConfig):
+def efficientnet(image_input: tf.keras.layers.Input, config: ModelConfig):  # pytype: disable=invalid-annotation  # typed-keras
  """Creates an EfficientNet graph given the model parameters.

  This function is wrapped by the `EfficientNet` class to make a tf.keras.Model.