silence-tensorflow 1.1.0

Simple python package to shut up Tensorflow warnings and logs.

Simple python package to shut up Tensorflow warnings and logs.

How do I install this package?

As usual, just download it using pip:

pip install silence_tensorflow

Tests Coverage

Since some software handling coverages sometimes get slightly different results, here’s three of them:

How do I use it?

You only need to import the package before import Tensorflow:

from silence_tensorflow import silence_tensorflow
silence_tensorflow()
import tensorflow as tf

...

# your code

How does this work under the hood?

This package will set the KMP_AFFINITY system variable to “noverbose”, TF_CPP_MIN_LOG_LEVEL to level 3 (only errors logged) and silence both FutureWarning and DeprecationWarning.

If you need a custom value for KMP_AFFINITY you should reset it after importing the package, as follows:

import os
from silence_tensorflow import silence_tensorflow
backup = os.environ["KMP_AFFINITY"]
silence_tensorflow()
os.environ["KMP_AFFINITY"] = backup