diff --git a/baselines/common/misc_util.py b/baselines/common/misc_util.py index 451de1c..6a296d4 100644 --- a/baselines/common/misc_util.py +++ b/baselines/common/misc_util.py @@ -76,10 +76,9 @@ def set_global_seeds(i): myseed = i + 1000 * rank if i is not None else None try: import tensorflow as tf + tf.set_random_seed(myseed) except ImportError: pass - else: - tf.set_random_seed(myseed) np.random.seed(myseed) random.seed(myseed)