Merge branch 'main' of into main-public

Author: ashahba

notebooks/question_answering/tfhub_question_answering/BERT_Question_Answering.ipynb

@@ -106,7 +106,7 @@
     "# Path where the https://github.com/tensorflow/models repo will be cloned\n",
     "tf_models_dir = os.path.join(output_directory, \"tensorflow-models\")\n",
     "os.environ[\"TF_MODELS_DIR\"] = tf_models_dir\n",
-    "tf_models_branch = \"v2.7.0\"\n",
+    "tf_models_branch = \"v2.11.0\"\n",
     "\n",
     "# Clone the TensorFlow models repo\n",
     "if not os.path.exists(tf_models_dir):\n",
@@ -296,7 +296,7 @@
     "    os.makedirs(checkpoint_dir)\n",
     "\n",
     "os.environ[\"TFHUB_CACHE_DIR\"] = os.path.join(output_directory, \"tfhub_modules\")\n",
-    "!python $tf_models_dir/official/nlp/bert/run_squad.py \\\n",
+    "!python $tf_models_dir/official/legacy/bert/run_squad.py \\\n",
     "  --mode=train_and_eval \\\n",
     "  --input_meta_data_path=$squad_metadata_path \\\n",
     "  --train_data_path=$train_mini_tfrecords_path \\\n",
@@ -337,9 +337,9 @@
    "outputs": [],
    "source": [
     "import tensorflow as tf\n",
-    "from official.nlp.bert import bert_models\n",
-    "from official.nlp.bert import configs as bert_configs\n",
-    "from official.nlp.bert import model_saving_utils\n",
+    "from official.legacy.bert import bert_models\n",
+    "from official.legacy.bert import configs as bert_configs\n",
+    "from official.legacy.bert import model_saving_utils\n",
     "\n",
     "tf.keras.mixed_precision.set_global_policy('float32')\n",
     "bert_config_obj = bert_configs.BertConfig.from_json_file(bert_config)\n",

notebooks/question_answering/tfhub_question_answering/bert_qa_utils.py

@@ -26,7 +26,7 @@
 sys.path.append(os.environ["TF_MODELS_DIR"])
 
 from official.common import distribute_utils
-from official.nlp.bert.run_squad_helper import get_dataset_fn
+from official.legacy.bert.run_squad_helper import get_dataset_fn
 from tlt.utils.file_utils import download_file
 from zipfile import ZipFile