Internal change

PiperOrigin-RevId: 432214649

official/projects/s3d/configs/s3d.py

@@ -18,8 +18,8 @@ import dataclasses
 from typing import Text
 
 from official.modeling import hyperparams
-from official.vision.beta.configs import backbones_3d
-from official.vision.beta.configs import video_classification
+from official.vision.configs import backbones_3d
+from official.vision.configs import video_classification
 
 
 @dataclasses.dataclass
@@ -97,4 +97,3 @@ class S3DModel(video_classification.VideoClassificationModel):
   """
   model_type: str = 's3d'
   backbone: Backbone3D = Backbone3D()
-

official/projects/s3d/modeling/inception_utils.py

@@ -19,7 +19,7 @@ from typing import Callable, Dict, Optional, Sequence, Set, Text, Tuple, Type, U
 import tensorflow as tf
 
 from official.projects.s3d.modeling import net_utils
-from official.vision.beta.modeling.layers import nn_blocks_3d
+from official.vision.modeling.layers import nn_blocks_3d
 
 INCEPTION_V1_CONV_ENDPOINTS = [
     'Conv2d_1a_7x7', 'Conv2d_2c_3x3', 'Mixed_3b', 'Mixed_3c', 'Mixed_4b',

official/projects/s3d/modeling/s3d.py

@@ -26,8 +26,8 @@ from official.modeling import hyperparams
 from official.projects.s3d.configs import s3d as cfg
 from official.projects.s3d.modeling import inception_utils
 from official.projects.s3d.modeling import net_utils
-from official.vision.beta.modeling import factory_3d as model_factory
-from official.vision.beta.modeling.backbones import factory as backbone_factory
+from official.vision.modeling import factory_3d as model_factory
+from official.vision.modeling.backbones import factory as backbone_factory
 
 initializers = tf.keras.initializers
 regularizers = tf.keras.regularizers

official/projects/s3d/train.py

@@ -17,16 +17,14 @@
 
 from absl import app
 
-# pylint: disable=unused-import
-from official.common import registry_imports
-# pylint: enable=unused-import
 from official.common import flags as tfm_flags
 # pylint: disable=unused-import
 from official.projects.s3d.configs.google import s3d as s3d_config
 from official.projects.s3d.modeling import s3d
 from official.projects.s3d.tasks.google import automl_video_classification
+from official.vision import registry_imports
 # pylint: enable=unused-import
-from official.vision.beta import train
+from official.vision import train
 
 if __name__ == '__main__':
   tfm_flags.define_flags()