Skip to content

Y
ygo-agent
Commit 63e558fb authored by sbl1996@126.com's avatar sbl1996@126.com
Browse files
Options

Adjust random seed

parent 38c7748d
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
scripts/cleanba.py
View file @ 63e558fb
...@@ -252,7 +252,7 @@ def rollout( ...@@ -252,7 +252,7 @@ def rollout(
if eval_mode != 'bot': if eval_mode != 'bot':
eval_params = params_queue.get() eval_params = params_queue.get()
local_seed = args.seed + device_thread_id local_seed = args.seed + device_thread_id * 100
np.random.seed(local_seed) np.random.seed(local_seed)
envs = make_env( envs = make_env(
...@@ -266,7 +266,7 @@ def rollout( ...@@ -266,7 +266,7 @@ def rollout(
eval_envs = make_env( eval_envs = make_env(
args, args,
local_seed, local_seed + 10000,
args.local_eval_episodes, args.local_eval_episodes,
args.local_eval_episodes // 4, mode=eval_mode, eval=True) args.local_eval_episodes // 4, mode=eval_mode, eval=True)
eval_envs = RecordEpisodeStatistics(eval_envs) eval_envs = RecordEpisodeStatistics(eval_envs)
...@@ -573,7 +573,7 @@ if __name__ == "__main__": ...@@ -573,7 +573,7 @@ if __name__ == "__main__":
args.ckpt_dir, save_fn, n_saved=2) args.ckpt_dir, save_fn, n_saved=2)
# seeding # seeding
seed_offset = args.local_rank * 10000 seed_offset = args.local_rank * 1000
args.seed += seed_offset args.seed += seed_offset
random.seed(args.seed) random.seed(args.seed)
init_key = jax.random.PRNGKey(args.seed - seed_offset) init_key = jax.random.PRNGKey(args.seed - seed_offset)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment