Internal change

PiperOrigin-RevId: 416593512

official/vision/beta/projects/assemblenet/configs/assemblenet.py → official/projects/assemblenet/configs/assemblenet.py

@@ -35,14 +35,14 @@ used for each layer in the inverted bottleneck modules.
 The structure_weights specify the learned connection weights.
 """
 import dataclasses
-from typing import List, Tuple, Optional
+from typing import List, Optional, Tuple
 
 from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.vision.beta.configs import backbones_3d
 from official.vision.beta.configs import common
-from official.vision.beta.configs import video_classification
+from official.vision.beta.configs.google import video_classification
 
 
 @dataclasses.dataclass

official/vision/beta/projects/assemblenet/configs/assemblenet_test.py → official/projects/assemblenet/configs/assemblenet_test.py

@@ -17,8 +17,8 @@ from absl.testing import parameterized
 import tensorflow as tf
 from official.core import config_definitions as cfg
 from official.core import exp_factory
+from official.projects.assemblenet.configs import assemblenet
 from official.vision.beta.configs import video_classification as exp_cfg
-from official.vision.beta.projects.assemblenet.configs import assemblenet
 
 
 class AssemblenetTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/assemblenet/modeling/assemblenet.py → official/projects/assemblenet/modeling/assemblenet.py

@@ -48,17 +48,17 @@ necessary. This is to run this on TPU efficiently.
 
 import functools
 import math
-from typing import Any, Mapping, List, Callable, Optional
+from typing import Any, Callable, List, Mapping, Optional
 
 from absl import logging
 import numpy as np
 import tensorflow as tf
 
 from official.modeling import hyperparams
+from official.projects.assemblenet.configs import assemblenet as cfg
+from official.projects.assemblenet.modeling import rep_flow_2d_layer as rf
 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.beta.projects.assemblenet.configs import assemblenet as cfg
-from official.vision.beta.projects.assemblenet.modeling import rep_flow_2d_layer as rf
 
 layers = tf.keras.layers
 intermediate_channel_size = [64, 128, 256, 512]

official/vision/beta/projects/assemblenet/modeling/assemblenet_plus.py → official/projects/assemblenet/modeling/assemblenet_plus.py

@@ -54,18 +54,18 @@ necessary. This is to run this on TPU efficiently.
 """
 
 import functools
-from typing import Any, Dict, Mapping, List, Optional
+from typing import Any, Dict, List, Mapping, Optional
 
 from absl import logging
 import numpy as np
 import tensorflow as tf
 
 from official.modeling import hyperparams
+from official.projects.assemblenet.configs import assemblenet as cfg
+from official.projects.assemblenet.modeling import assemblenet as asn
+from official.projects.assemblenet.modeling import rep_flow_2d_layer as rf
 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.beta.projects.assemblenet.configs import assemblenet as cfg
-from official.vision.beta.projects.assemblenet.modeling import assemblenet as asn
-from official.vision.beta.projects.assemblenet.modeling import rep_flow_2d_layer as rf
 
 layers = tf.keras.layers
 

official/vision/beta/projects/assemblenet/modeling/assemblenet_plus_test.py → official/projects/assemblenet/modeling/assemblenet_plus_test.py

@@ -18,8 +18,8 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from official.vision.beta.projects.assemblenet.configs import assemblenet as asn_config
-from official.vision.beta.projects.assemblenet.modeling import assemblenet_plus as asnp
+from official.projects.assemblenet.configs import assemblenet as asn_config
+from official.projects.assemblenet.modeling import assemblenet_plus as asnp
 
 
 class AssembleNetPlusTest(parameterized.TestCase, tf.test.TestCase):

official/vision/beta/projects/assemblenet/train.py → official/projects/assemblenet/train.py

@@ -39,9 +39,9 @@ from official.core import train_lib
 from official.core import train_utils
 from official.modeling import performance
 # pylint: disable=unused-import
-from official.vision.beta.projects.assemblenet.configs import assemblenet as asn_configs
-from official.vision.beta.projects.assemblenet.modeling import assemblenet as asn
-from official.vision.beta.projects.assemblenet.modeling import assemblenet_plus as asnp
+from official.projects.assemblenet.configs import assemblenet as asn_configs
+from official.projects.assemblenet.modeling import assemblenet as asn
+from official.projects.assemblenet.modeling import assemblenet_plus as asnp
 # pylint: enable=unused-import
 
 FLAGS = flags.FLAGS

official/vision/beta/projects/assemblenet/train_test.py → official/projects/assemblenet/train_test.py

@@ -21,9 +21,8 @@ from absl import flags
 from absl import logging
 from absl.testing import flagsaver
 import tensorflow as tf
-
+from official.projects.assemblenet import train as train_lib
 from official.vision.beta.dataloaders import tfexample_utils
-from official.vision.beta.projects.assemblenet import train as train_lib
 
 FLAGS = flags.FLAGS