diff --git a/baselines/common/misc_util.py b/baselines/common/misc_util.py index 48bc3dedc5..96ab3cad94 100644 --- a/baselines/common/misc_util.py +++ b/baselines/common/misc_util.py @@ -55,7 +55,7 @@ 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) + tf.random.set_seed(myseed) except ImportError: pass np.random.seed(myseed)