From 89b1450df72aa62d9f148c189393244d8e767bfc Mon Sep 17 00:00:00 2001 From: Shangtong Zhang Date: Mon, 28 May 2018 15:27:42 -0600 Subject: [PATCH] Force random seed --- deep_rl/utils/misc.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/deep_rl/utils/misc.py b/deep_rl/utils/misc.py index a4aa07c..4f8e0d0 100644 --- a/deep_rl/utils/misc.py +++ b/deep_rl/utils/misc.py @@ -16,7 +16,12 @@ except: # python == 2.7 from pathlib2 import Path +def random_seed(): + np.random.seed() + torch.manual_seed(np.random.randint(int(1e6))) + def run_episodes(agent): + random_seed() config = agent.config window_size = 100 ep = 0 @@ -49,6 +54,7 @@ def run_episodes(agent): return steps, rewards, avg_test_rewards def run_iterations(agent): + random_seed() config = agent.config agent_name = agent.__class__.__name__ iteration = 0