Internal change

PiperOrigin-RevId: 422638136

official/legacy/albert/configs.py

@@ -16,7 +16,7 @@
 
 import six
 
-from official.nlp.bert import configs
+from official.legacy.bert import configs
 
 
 class AlbertConfig(configs.BertConfig):

official/nlp/bert/README.md → official/legacy/bert/README.md

@@ -2,7 +2,7 @@
 
 **WARNING**: We are on the way to deprecate most of the code in this directory.
 Please see
-[this link](https://github.com/tensorflow/models/blob/master/official/nlp/docs/train.md)
+[this link](../g3doc/tutorials/bert_new.md)
 for the new tutorial and use the new code in `nlp/modeling`. This README is
 still correct for this legacy implementation.
 

official/nlp/bert/bert_models.py → official/legacy/bert/bert_models.py

@@ -18,8 +18,8 @@ import gin
 import tensorflow as tf
 import tensorflow_hub as hub
 from official.legacy.albert import configs as albert_configs
+from official.legacy.bert import configs
 from official.modeling import tf_utils
-from official.nlp.bert import configs
 from official.nlp.modeling import models
 from official.nlp.modeling import networks
 

official/nlp/bert/bert_models_test.py → official/legacy/bert/bert_models_test.py

@@ -14,8 +14,8 @@
 
 import tensorflow as tf
 
-from official.nlp.bert import bert_models
-from official.nlp.bert import configs as bert_configs
+from official.legacy.bert import bert_models
+from official.legacy.bert import configs as bert_configs
 from official.nlp.modeling import networks
 
 

official/nlp/bert/export_tfhub.py → official/legacy/bert/export_tfhub.py

@@ -25,8 +25,8 @@ from absl import app
 from absl import flags
 from absl import logging
 import tensorflow as tf
-from official.nlp.bert import bert_models
-from official.nlp.bert import configs
+from official.legacy.bert import bert_models
+from official.legacy.bert import configs
 
 FLAGS = flags.FLAGS
 

official/nlp/bert/export_tfhub_test.py → official/legacy/bert/export_tfhub_test.py

@@ -21,8 +21,8 @@ import numpy as np
 import tensorflow as tf
 import tensorflow_hub as hub
 
-from official.nlp.bert import configs
-from official.nlp.bert import export_tfhub
+from official.legacy.bert import configs
+from official.legacy.bert import export_tfhub
 
 
 class ExportTfhubTest(tf.test.TestCase, parameterized.TestCase):

official/nlp/bert/model_saving_utils.py → official/legacy/bert/model_saving_utils.py

@@ -15,10 +15,9 @@
 """Utilities to save models."""
 
 import os
-
+import typing
 from absl import logging
 import tensorflow as tf
-import typing
 
 
 def export_bert_model(model_export_path: typing.Text,

official/nlp/bert/model_training_utils_test.py → official/legacy/bert/model_training_utils_test.py

@@ -25,8 +25,8 @@ import tensorflow as tf
 
 from tensorflow.python.distribute import combinations
 from tensorflow.python.distribute import strategy_combinations
-from official.nlp.bert import common_flags
-from official.nlp.bert import model_training_utils
+from official.legacy.bert import common_flags
+from official.legacy.bert import model_training_utils
 
 
 common_flags.define_common_bert_flags()

official/nlp/bert/run_classifier.py → official/legacy/bert/run_classifier.py

@@ -26,13 +26,13 @@ from absl import logging
 import gin
 import tensorflow as tf
 from official.common import distribute_utils
+from official.legacy.bert import bert_models
+from official.legacy.bert import common_flags
+from official.legacy.bert import configs as bert_configs
+from official.legacy.bert import input_pipeline
+from official.legacy.bert import model_saving_utils
 from official.modeling import performance
 from official.nlp import optimization
-from official.nlp.bert import bert_models
-from official.nlp.bert import common_flags
-from official.nlp.bert import configs as bert_configs
-from official.nlp.bert import input_pipeline
-from official.nlp.bert import model_saving_utils
 from official.utils.misc import keras_utils
 
 flags.DEFINE_enum(

official/nlp/bert/run_pretraining.py → official/legacy/bert/run_pretraining.py

@@ -21,13 +21,13 @@ from absl import logging
 import gin
 import tensorflow as tf
 from official.common import distribute_utils
+from official.legacy.bert import bert_models
+from official.legacy.bert import common_flags
+from official.legacy.bert import configs
+from official.legacy.bert import input_pipeline
+from official.legacy.bert import model_training_utils
 from official.modeling import performance
 from official.nlp import optimization
-from official.nlp.bert import bert_models
-from official.nlp.bert import common_flags
-from official.nlp.bert import configs
-from official.nlp.bert import input_pipeline
-from official.nlp.bert import model_training_utils
 
 
 flags.DEFINE_string('input_files', None,

official/nlp/bert/run_squad.py → official/legacy/bert/run_squad.py

@@ -25,10 +25,10 @@ from absl import logging
 import gin
 import tensorflow as tf
 from official.common import distribute_utils
-from official.nlp.bert import configs as bert_configs
-from official.nlp.bert import run_squad_helper
-from official.nlp.bert import tokenization
+from official.legacy.bert import configs as bert_configs
+from official.legacy.bert import run_squad_helper
 from official.nlp.data import squad_lib as squad_lib_wp
+from official.nlp.tools import tokenization
 from official.utils.misc import keras_utils
 
 

official/nlp/bert/run_squad_helper.py → official/legacy/bert/run_squad_helper.py

@@ -21,16 +21,16 @@ import os
 from absl import flags
 from absl import logging
 import tensorflow as tf
+from official.legacy.bert import bert_models
+from official.legacy.bert import common_flags
+from official.legacy.bert import input_pipeline
+from official.legacy.bert import model_saving_utils
+from official.legacy.bert import model_training_utils
 from official.modeling import performance
 from official.nlp import optimization
-from official.nlp.bert import bert_models
-from official.nlp.bert import common_flags
-from official.nlp.bert import input_pipeline
-from official.nlp.bert import model_saving_utils
-from official.nlp.bert import model_training_utils
-from official.nlp.bert import squad_evaluate_v1_1
-from official.nlp.bert import squad_evaluate_v2_0
 from official.nlp.data import squad_lib_sp
+from official.nlp.tools import squad_evaluate_v1_1
+from official.nlp.tools import squad_evaluate_v2_0
 from official.utils.misc import keras_utils
 
 

official/nlp/bert/serving.py → official/legacy/bert/serving.py

@@ -18,8 +18,8 @@ from absl import app
 from absl import flags
 import tensorflow as tf
 
-from official.nlp.bert import bert_models
-from official.nlp.bert import configs
+from official.legacy.bert import bert_models
+from official.legacy.bert import configs
 
 flags.DEFINE_integer(
     "sequence_length", None, "Sequence length to parse the tf.Example. If "

official/nlp/data/classifier_data_lib.py

@@ -24,7 +24,7 @@ from absl import logging
 import tensorflow as tf
 import tensorflow_datasets as tfds
 
-from official.nlp.bert import tokenization
+from official.nlp.tools import tokenization
 
 
 class InputExample(object):