Internal change

PiperOrigin-RevId: 473377893

official/vision/beta/projects/simclr/README.md → official/projects/simclr/README.md

@@ -25,7 +25,7 @@ install -r ./official/requirements.txt`
 To pretrain the model on Imagenet, try the following command:
 
 ```
-python3 -m official.vision.beta.projects.simclr.train \
+python3 -m official.projects.simclr.train \
   --mode=train_and_eval \
   --experiment=simclr_pretraining \
   --model_dir={MODEL_DIR} \
@@ -44,7 +44,7 @@ You can also find image IDs of these subsets in `imagenet_subsets/`.
 To fine-tune the whole network, refer to the following command:
 
 ```
-python3 -m official.vision.beta.projects.simclr.train \
+python3 -m official.projects.simclr.train \
   --mode=train_and_eval \
   --experiment=simclr_finetuning \
   --model_dir={MODEL_DIR} \

official/vision/beta/projects/simclr/common/registry_imports.py → official/projects/simclr/common/registry_imports.py

@@ -15,8 +15,8 @@
 """All necessary imports for registration."""
 
 # pylint: disable=unused-import
+from official.projects.simclr.configs import simclr
+from official.projects.simclr.losses import contrastive_losses
+from official.projects.simclr.modeling import simclr_model
+from official.projects.simclr.tasks import simclr as simclr_task
 from official.vision import registry_imports
-from official.vision.beta.projects.simclr.configs import simclr
-from official.vision.beta.projects.simclr.losses import contrastive_losses
-from official.vision.beta.projects.simclr.modeling import simclr_model
-from official.vision.beta.projects.simclr.tasks import simclr as simclr_task

official/vision/beta/projects/simclr/configs/multitask_config.py → official/projects/simclr/configs/multitask_config.py

@@ -20,8 +20,8 @@ from typing import List, Tuple
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.modeling.multitask import configs as multitask_configs
-from official.vision.beta.projects.simclr.configs import simclr as simclr_configs
-from official.vision.beta.projects.simclr.modeling import simclr_model
+from official.projects.simclr.configs import simclr as simclr_configs
+from official.projects.simclr.modeling import simclr_model
 from official.vision.configs import backbones
 from official.vision.configs import common
 

official/vision/beta/projects/simclr/configs/multitask_config_test.py → official/projects/simclr/configs/multitask_config_test.py

@@ -18,8 +18,8 @@ import tensorflow as tf
 
 from official.core import exp_factory
 from official.modeling.multitask import configs as multitask_configs
-from official.vision.beta.projects.simclr.configs import multitask_config as simclr_multitask_config
-from official.vision.beta.projects.simclr.configs import simclr as exp_cfg
+from official.projects.simclr.configs import multitask_config as simclr_multitask_config
+from official.projects.simclr.configs import simclr as exp_cfg
 
 
 class MultitaskConfigTest(tf.test.TestCase):

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

@@ -21,7 +21,7 @@ from official.core import config_definitions as cfg
 from official.core import exp_factory
 from official.modeling import hyperparams
 from official.modeling import optimization
-from official.vision.beta.projects.simclr.modeling import simclr_model
+from official.projects.simclr.modeling import simclr_model
 from official.vision.configs import backbones
 from official.vision.configs import common
 

official/vision/beta/projects/simclr/configs/simclr_test.py → official/projects/simclr/configs/simclr_test.py

@@ -19,8 +19,8 @@ import tensorflow as tf
 
 from official.core import config_definitions as cfg
 from official.core import exp_factory
-from official.vision.beta.projects.simclr.common import registry_imports  # pylint: disable=unused-import
-from official.vision.beta.projects.simclr.configs import simclr as exp_cfg
+from official.projects.simclr.common import registry_imports  # pylint: disable=unused-import
+from official.projects.simclr.configs import simclr as exp_cfg
 
 
 class SimCLRConfigTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/simclr/dataloaders/simclr_input.py → official/projects/simclr/dataloaders/simclr_input.py

@@ -40,8 +40,8 @@ from typing import List
 
 import tensorflow as tf
 
-from official.vision.beta.projects.simclr.dataloaders import preprocess_ops as simclr_preprocess_ops
-from official.vision.beta.projects.simclr.modeling import simclr_model
+from official.projects.simclr.dataloaders import preprocess_ops as simclr_preprocess_ops
+from official.projects.simclr.modeling import simclr_model
 from official.vision.dataloaders import decoder
 from official.vision.dataloaders import parser
 from official.vision.ops import preprocess_ops

official/vision/beta/projects/simclr/heads/simclr_head.py → official/projects/simclr/heads/simclr_head.py

@@ -14,11 +14,11 @@
 
 """SimCLR prediction heads."""
 
-from typing import Text, Optional
+from typing import Optional, Text
 
 import tensorflow as tf
 
-from official.vision.beta.projects.simclr.modeling.layers import nn_blocks
+from official.projects.simclr.modeling.layers import nn_blocks
 
 regularizers = tf.keras.regularizers
 layers = tf.keras.layers

official/vision/beta/projects/simclr/heads/simclr_head_test.py → official/projects/simclr/heads/simclr_head_test.py

@@ -17,7 +17,7 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from official.vision.beta.projects.simclr.heads import simclr_head
+from official.projects.simclr.heads import simclr_head
 
 
 class ProjectionHeadTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/simclr/losses/contrastive_losses_test.py → official/projects/simclr/losses/contrastive_losses_test.py

@@ -17,7 +17,7 @@ from absl.testing import parameterized
 import numpy as np
 import tensorflow as tf
 
-from official.vision.beta.projects.simclr.losses import contrastive_losses
+from official.projects.simclr.losses import contrastive_losses
 
 
 class ContrastiveLossesTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/simclr/modeling/layers/nn_blocks_test.py → official/projects/simclr/modeling/layers/nn_blocks_test.py

@@ -16,7 +16,7 @@ from absl.testing import parameterized
 
 import tensorflow as tf
 
-from official.vision.beta.projects.simclr.modeling.layers import nn_blocks
+from official.projects.simclr.modeling.layers import nn_blocks
 
 
 class DenseBNTest(tf.test.TestCase, parameterized.TestCase):