change TF console log level to 'errors only'

2019-03-09 20:23:33 +04:00
parent 95e70d510f
commit 9440224556
1 changed files with 1 additions and 1 deletions
--- a/nnlib/nnlib.py
+++ b/nnlib/nnlib.py
@@ -121,7 +121,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
            os.environ.pop('CUDA_VISIBLE_DEVICES')
        
        os.environ['TF_MIN_GPU_MULTIPROCESSOR_COUNT'] = '2'
-        
+        os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' #tf log errors only
        import tensorflow as tf
        nnlib.tf = tf