From 751e2460876930ce29f15329a765ca049d323e36 Mon Sep 17 00:00:00 2001 From: thomwolf Date: Thu, 10 Oct 2019 15:47:20 +0200 Subject: [PATCH] using tf.print in roberta --- transformers/modeling_tf_roberta.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformers/modeling_tf_roberta.py b/transformers/modeling_tf_roberta.py index db62dd3014..83c1db0495 100644 --- a/transformers/modeling_tf_roberta.py +++ b/transformers/modeling_tf_roberta.py @@ -74,7 +74,7 @@ class TFRobertaMainLayer(TFBertMainLayer): input_ids = inputs if tf.not_equal(tf.reduce_sum(input_ids[:, 0]), 0): - logger.warning("A sequence with no special tokens has been passed to the RoBERTa model. " + tf.print("A sequence with no special tokens has been passed to the RoBERTa model. " "This model requires special tokens in order to work. " "Please specify add_special_tokens=True in your encoding.")