Commit c6f0446d authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 447902944
parent a281aaa2
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
This is only a dummy example to showcase how a model is composed. It is usually This is only a dummy example to showcase how a model is composed. It is usually
not needed to implement a modedl from scratch. Most SoTA models can be found and not needed to implement a modedl from scratch. Most SoTA models can be found and
directly used from `official/vision/beta/modeling` directory. directly used from `official/vision/modeling` directory.
""" """
from typing import Any, Mapping from typing import Any, Mapping
...@@ -84,7 +84,7 @@ def build_example_model(input_specs: tf.keras.layers.InputSpec, ...@@ -84,7 +84,7 @@ def build_example_model(input_specs: tf.keras.layers.InputSpec,
This function is the main entry point to build a model. Commonly, it build a This function is the main entry point to build a model. Commonly, it build a
model by building a backbone, decoder and head. An example of building a model by building a backbone, decoder and head. An example of building a
classification model is at classification model is at
third_party/tensorflow_models/official/vision/beta/modeling/backbones/resnet.py. third_party/tensorflow_models/official/vision/modeling/backbones/resnet.py.
However, it is not mandatory for all models to have these three pieces However, it is not mandatory for all models to have these three pieces
exactly. Depending on the task, model can be as simple as the example model exactly. Depending on the task, model can be as simple as the example model
here or more complex, such as multi-head architecture. here or more complex, such as multi-head architecture.
......
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