Skip to content

GitLab

Commit 24b2ccd4 authored by Rebecca Chen's avatar Rebecca Chen Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 398612564
parent e5fbe328
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/image_classification/efficientnet/efficientnet_model.py
View file @ 24b2ccd4
...@@ -327,7 +327,7 @@ def mb_conv_block(inputs: tf.Tensor, ...@@ -327,7 +327,7 @@ def mb_conv_block(inputs: tf.Tensor,
return x 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. """Creates an EfficientNet graph given the model parameters.
This function is wrapped by the `EfficientNet` class to make a tf.keras.Model. This function is wrapped by the `EfficientNet` class to make a tf.keras.Model.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment