Python hugging face warning

Question

my code is working but I am getting this warning and how can I avoid it

All model checkpoint layers were used when initializing TFRobertaForSequenceClassification.

All the layers of TFRobertaForSequenceClassification were initialized from the model checkpoint at arpanghoshal/EmoRoBERTa.
If your task is similar to the task the model of the checkpoint was trained on, you can already use TFRobertaForSequenceClassification for predictions without further training.

Nebiim

Please provide enough code so others can better understand or reproduce the problem. – Windpipe

Answer 1

You can manage the warnings with the logging utility introduced in version 3.1.0:

from transformers import logging

logging.set_verbosity_warning()

Araucania

Still getting that message – Nebiim

You actually want to set the log level to error: logging.set_verbosity_error(). @Nebiim – Costmary

Answer 2

The code below suppresses warning messages:

from transformers.utils import logging
logging.set_verbosity_error() 

Shipwright

Answer 3

For me, importing solution was not working, as the warning message was logged during the import itself. What helped me was to set up the environment variable before the import:

os.environ["TRANSFORMERS_VERBOSITY"] = "error"

Compassionate