remove logging

tensorflow · May 4, 2024 · 66cf0c0 · 66cf0c0
1 parent eabe1e8
commit 66cf0c0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/official/recommendation/ranking/data/data_pipeline_multi_hot.py b/official/recommendation/ranking/data/data_pipeline_multi_hot.py
@@ -43,7 +43,7 @@ def __init__(self,
     self.sparse_features = ['categorical-feature-%d' % x for x in range(14, 40)]
 
   def __call__(self, ctx: tf.distribute.InputContext):
-    tf._logging.info(f'file pattern: {self._file_pattern}')
+
     params = self._params
     # Per replica batch size.
     batch_size = (

diff --git a/official/recommendation/ranking/train.py b/official/recommendation/ranking/train.py
@@ -69,10 +69,6 @@ def main(_) -> None:
     # may race against the train job for writing the same file.
     train_utils.serialize_config(params, model_dir)
 
-  if FLAGS.seed is not None:
-    logging.info('Setting tf seed.')
-    tf.random.set_seed(FLAGS.seed)
-
   task = RankingTask(
       params=params.task,
       trainer_config=params.trainer,