diff --git a/baselines/ddpg/ddpg.py b/baselines/ddpg/ddpg.py index 181f923..8b8659b 100755 --- a/baselines/ddpg/ddpg.py +++ b/baselines/ddpg/ddpg.py @@ -7,7 +7,7 @@ from baselines.ddpg.ddpg_learner import DDPG from baselines.ddpg.models import Actor, Critic from baselines.ddpg.memory import Memory from baselines.ddpg.noise import AdaptiveParamNoiseSpec, NormalActionNoise, OrnsteinUhlenbeckActionNoise - +from baselines.common import set_global_seeds import baselines.common.tf_util as U from baselines import logger @@ -41,6 +41,7 @@ def learn(network, env, param_noise_adaption_interval=50, **network_kwargs): + set_global_seeds(seed) if total_timesteps is not None: assert nb_epochs is None