filter erroneos TF warning

apax-hub · Mar 29, 2024 · 07c86b5 · 07c86b5
1 parent 90c9f69
commit 07c86b5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/apax/__init__.py b/apax/__init__.py
@@ -3,6 +3,7 @@
 import jax
 
 os.environ["XLA_PYTHON_CLIENT_PREALLOCATE"] = "false"
+os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 
 jax.config.update("jax_enable_x64", True)
 from apax.utils.helpers import setup_ase