diff --git a/tests/memory/prioritized_test.py b/tests/memory/prioritized_test.py index 957badd..254372b 100644 --- a/tests/memory/prioritized_test.py +++ b/tests/memory/prioritized_test.py @@ -4,6 +4,7 @@ import numpy as np import random import tensorflow as tf +tf.logging.set_verbosity(tf.logging.ERROR) class PrioritizedMemoryTest(tf.test.TestCase): diff --git a/tests/memory/simple_test.py b/tests/memory/simple_test.py index 46d14a5..b28a1ca 100644 --- a/tests/memory/simple_test.py +++ b/tests/memory/simple_test.py @@ -4,6 +4,7 @@ import numpy as np import random import tensorflow as tf +tf.logging.set_verbosity(tf.logging.ERROR) class SimpleMemoryTest(tf.test.TestCase): diff --git a/tests/models/basic_test.py b/tests/models/basic_test.py index e2e10dd..19215ad 100644 --- a/tests/models/basic_test.py +++ b/tests/models/basic_test.py @@ -3,6 +3,7 @@ import numpy as np import tensorflow as tf +tf.logging.set_verbosity(tf.logging.ERROR) inp_dim = 80 diff --git a/tests/models/ddpg_test.py b/tests/models/ddpg_test.py index fa1ce93..c522611 100644 --- a/tests/models/ddpg_test.py +++ b/tests/models/ddpg_test.py @@ -3,6 +3,7 @@ import numpy as np import tensorflow as tf +tf.logging.set_verbosity(tf.logging.ERROR) inp_dim = 80 diff --git a/tests/models/noisy_network_test.py b/tests/models/noisy_network_test.py index 1d7eb81..2e44e0d 100644 --- a/tests/models/noisy_network_test.py +++ b/tests/models/noisy_network_test.py @@ -3,6 +3,7 @@ import numpy as np import tensorflow as tf +tf.logging.set_verbosity(tf.logging.ERROR) inp_dim = 80 diff --git a/tests/models/ppo_test.py b/tests/models/ppo_test.py index 40abbf2..43ff1b5 100644 --- a/tests/models/ppo_test.py +++ b/tests/models/ppo_test.py @@ -3,6 +3,7 @@ import numpy as np import tensorflow as tf +tf.logging.set_verbosity(tf.logging.ERROR) inp_dim = 80 diff --git a/train.py b/train.py index 2595aa7..78a1fbf 100644 --- a/train.py +++ b/train.py @@ -4,6 +4,7 @@ import numpy as np import tensorflow as tf import tensorflow.keras.backend as K +tf.logging.set_verbosity(tf.logging.ERROR) from rl.utils import flags from rl.utils.utils import ModeKeys