Commit de44470f authored by Xianzhi Du's avatar Xianzhi Du Committed by A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 423333566
parent d3a3f143
...@@ -15,4 +15,4 @@ ...@@ -15,4 +15,4 @@
# Lint as: python3 # Lint as: python3
"""Configs package definition.""" """Configs package definition."""
from official.vision.beta.projects.vit.configs import image_classification from official.projects.vit.configs import image_classification
...@@ -26,7 +26,7 @@ from official.modeling import hyperparams ...@@ -26,7 +26,7 @@ from official.modeling import hyperparams
from official.modeling import optimization from official.modeling import optimization
from official.vision.beta.configs import common from official.vision.beta.configs import common
from official.vision.beta.configs import image_classification as img_cls_cfg from official.vision.beta.configs import image_classification as img_cls_cfg
from official.vision.beta.projects.vit.configs import backbones from official.projects.vit.configs import backbones
from official.vision.beta.tasks import image_classification from official.vision.beta.tasks import image_classification
DataConfig = img_cls_cfg.DataConfig DataConfig = img_cls_cfg.DataConfig
......
...@@ -17,9 +17,9 @@ ...@@ -17,9 +17,9 @@
import tensorflow as tf import tensorflow as tf
from official.modeling import activations from official.modeling import activations
from official.projects.vit.modeling import nn_blocks
from official.vision.beta.modeling.backbones import factory from official.vision.beta.modeling.backbones import factory
from official.vision.beta.modeling.layers import nn_layers from official.vision.beta.modeling.layers import nn_layers
from official.vision.beta.projects.vit.modeling import nn_blocks
layers = tf.keras.layers layers = tf.keras.layers
......
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
from absl.testing import parameterized from absl.testing import parameterized
import tensorflow as tf import tensorflow as tf
from official.vision.beta.projects.vit.modeling import vit from official.projects.vit.modeling import vit
class VisionTransformerTest(parameterized.TestCase, tf.test.TestCase): class VisionTransformerTest(parameterized.TestCase, tf.test.TestCase):
......
...@@ -18,9 +18,9 @@ ...@@ -18,9 +18,9 @@
from absl import app from absl import app
from official.common import flags as tfm_flags from official.common import flags as tfm_flags
from official.projects.vit import configs # pylint: disable=unused-import
from official.projects.vit.modeling import vit # pylint: disable=unused-import
from official.vision.beta import train from official.vision.beta import train
from official.vision.beta.projects.vit import configs # pylint: disable=unused-import
from official.vision.beta.projects.vit.modeling import vit # pylint: disable=unused-import
if __name__ == '__main__': if __name__ == '__main__':
......
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