From f68d22850ced09bb194b30068ff94ca3409f0879 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Thu, 2 Apr 2020 21:49:34 +0200 Subject: [PATCH] delete bogus print statement (#3595) --- src/transformers/modeling_tf_utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py index ab084f4ad2..fc75984c8b 100644 --- a/src/transformers/modeling_tf_utils.py +++ b/src/transformers/modeling_tf_utils.py @@ -1177,8 +1177,6 @@ class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin): beam_tokens = tf.convert_to_tensor([x[1] for x in next_batch_beam], dtype=tf.int32) beam_idx = tf.convert_to_tensor([x[2] for x in next_batch_beam], dtype=tf.int32) - print("Scores: {}-{}".format(cur_len, beam_scores.numpy())) - # re-order batch input_ids = tf.stack([tf.identity(input_ids[x, :]) for x in beam_idx]) input_ids = tf.concat([input_ids, tf.expand_dims(beam_tokens, 1)], axis=-1)