Internal change

PiperOrigin-RevId: 433284777

official/projects/vit/configs/image_classification.py

@@ -24,10 +24,10 @@ from official.core import exp_factory
 from official.core import task_factory
 from official.modeling import hyperparams
 from official.modeling import optimization
-from official.vision.beta.configs import common
-from official.vision.beta.configs import image_classification as img_cls_cfg
+from official.vision.configs import common
+from official.vision.configs import image_classification as img_cls_cfg
 from official.projects.vit.configs import backbones
-from official.vision.beta.tasks import image_classification
+from official.vision.tasks import image_classification
 
 DataConfig = img_cls_cfg.DataConfig
 

official/projects/vit/modeling/nn_blocks.py

@@ -16,7 +16,7 @@
 import tensorflow as tf
 
 from official.nlp import modeling
-from official.vision.beta.modeling.layers.nn_layers import StochasticDepth
+from official.vision.modeling.layers.nn_layers import StochasticDepth
 
 
 class TransformerEncoderBlock(modeling.layers.TransformerEncoderBlock):

official/projects/vit/modeling/vit.py

@@ -18,8 +18,8 @@ import tensorflow as tf
 
 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.layers import nn_layers
+from official.vision.modeling.backbones import factory
+from official.vision.modeling.layers import nn_layers
 
 
 layers = tf.keras.layers

official/projects/vit/train.py

@@ -20,7 +20,7 @@ from absl import app
 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 import train
 
 
 if __name__ == '__main__':