Internal change

PiperOrigin-RevId: 312096778

official/nlp/albert/run_classifier.py

@@ -32,8 +32,6 @@ FLAGS = flags.FLAGS
 
 
 def main(_):
-  # Users should always run this script under TF 2.x
-
   with tf.io.gfile.GFile(FLAGS.input_meta_data_path, 'rb') as reader:
     input_meta_data = json.loads(reader.read().decode('utf-8'))
 

official/nlp/albert/run_squad.py

@@ -94,8 +94,6 @@ def export_squad(model_export_path, input_meta_data):
 
 
 def main(_):
-  # Users should always run this script under TF 2.x
-
   with tf.io.gfile.GFile(FLAGS.input_meta_data_path, 'rb') as reader:
     input_meta_data = json.loads(reader.read().decode('utf-8'))
 

official/nlp/bert/run_classifier.py

@@ -434,7 +434,6 @@ def custom_main(custom_callbacks=None):
 
 
 def main(_):
-  # Users should always run this script under TF 2.x
   custom_main(custom_callbacks=None)
 
 

official/nlp/bert/run_pretraining.py

@@ -180,7 +180,6 @@ def run_bert_pretrain(strategy, custom_callbacks=None):
 
 
 def main(_):
-  # Users should always run this script under TF 2.x
   gin.parse_config_files_and_bindings(FLAGS.gin_file, FLAGS.gin_param)
   if not FLAGS.model_dir:
     FLAGS.model_dir = '/tmp/bert20/'

official/nlp/bert/run_squad.py

@@ -91,7 +91,6 @@ def export_squad(model_export_path, input_meta_data):
 
 
 def main(_):
-  # Users should always run this script under TF 2.x
   gin.parse_config_files_and_bindings(FLAGS.gin_file, FLAGS.gin_param)
 
   with tf.io.gfile.GFile(FLAGS.input_meta_data_path, 'rb') as reader: