Add registry_import file to include all necessary imports for MaxViT.

PiperOrigin-RevId: 533202307

official/projects/maxvit/configs/__init__.py

@@ -19,4 +19,4 @@ from official.projects.maxvit.configs import backbones  # pylint:disable=unused-
 from official.projects.maxvit.configs import rcnn  # pylint:disable=unused-import
 from official.projects.maxvit.configs import retinanet  # pylint:disable=unused-import
 from official.projects.maxvit.configs import semantic_segmentation  # pylint:disable=unused-import
-from official.projects.maxvit.configs.google import image_classification  # pylint:disable=unused-import
+from official.projects.maxvit.configs import image_classification  # pylint:disable=unused-import

official/projects/maxvit/registry_imports.py

+# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""All necessary imports for registration."""
+
+# pylint: disable=unused-import
+# pylint: disable=g-bad-import-order
+from official.vision import registry_imports
+from official.projects.maxvit import configs  # pylint: disable=unused-import
+from official.projects.maxvit.modeling import maxvit  # pylint: disable=unused-import

official/projects/maxvit/train.py

@@ -12,12 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""TensorFlow Model Garden Vision training driver, including ViT configs.."""
+"""TensorFlow Model Garden Vision training driver, including MaxViT configs.."""
 from absl import app
 
 from official.common import flags as tfm_flags
-from official.projects.maxvit import configs  # pylint: disable=unused-import
-from official.projects.maxvit.modeling import maxvit  # pylint: disable=unused-import
+from official.projects.maxvit import registry_imports  # pylint: disable=unused-import
 from official.vision import train