Remove hooks.

PiperOrigin-RevId: 306994199

official/benchmark/models/shakespeare/shakespeare_main.py

@@ -47,7 +47,6 @@ def define_flags():
                          epochs_between_evals=False,
                          stop_threshold=False,
                          num_gpu=True,
-                         hooks=False,
                          export_dir=False,
                          run_eagerly=True,
                          distribution_strategy=True)

official/nlp/bert/common_flags.py

@@ -39,7 +39,6 @@ def define_common_bert_flags():
       stop_threshold=False,
       batch_size=False,
       num_gpu=True,
-      hooks=False,
       export_dir=False,
       distribution_strategy=True,
       run_eagerly=True)

official/recommendation/ncf_common.py

@@ -149,7 +149,7 @@ def define_ncf_flags():
   flags_core.define_base(model_dir=True, clean=True, train_epochs=True,
                          epochs_between_evals=True, export_dir=False,
                          run_eagerly=True, stop_threshold=True, num_gpu=True,
-                         hooks=True, distribution_strategy=True)
+                         distribution_strategy=True)
   flags_core.define_performance(
       synthetic_data=True,
       dtype=True,
@@ -168,7 +168,6 @@ def define_ncf_flags():
       data_dir="/tmp/movielens-data/",
       train_epochs=2,
       batch_size=99000,
-      hooks="ProfilerHook",
       tpu=None
   )
 

official/recommendation/neumf_model.py

@@ -126,7 +126,6 @@ def neumf_model_fn(features, labels, mode, params):
         weights=tf.cast(valid_pt_mask, tf.float32)
     )
 
-    # This tensor is used by logging hooks.
     tf.identity(loss, name="cross_entropy")
 
     global_step = tf.compat.v1.train.get_global_step()