change TF console log level to 'errors only'
This commit is contained in:
+1
-1
@@ -121,7 +121,7 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
|
|||||||
os.environ.pop('CUDA_VISIBLE_DEVICES')
|
os.environ.pop('CUDA_VISIBLE_DEVICES')
|
||||||
|
|
||||||
os.environ['TF_MIN_GPU_MULTIPROCESSOR_COUNT'] = '2'
|
os.environ['TF_MIN_GPU_MULTIPROCESSOR_COUNT'] = '2'
|
||||||
|
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' #tf log errors only
|
||||||
import tensorflow as tf
|
import tensorflow as tf
|
||||||
nnlib.tf = tf
|
nnlib.tf = tf
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user