From 2a224726d6995913cb1caebaf654bb17aee736d0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alexandre=20Pich=C3=A9?= Date: Tue, 20 Oct 2020 10:28:27 -0400 Subject: [PATCH] Default number of ensemble Hi, I believe the default value for the number of ensemble should be more than 1. Otherwise the default agent does not solve Deep Sea. I set it to 20 to be consistent with the tensorflow implementation. Best. --- bsuite/baselines/jax/boot_dqn/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bsuite/baselines/jax/boot_dqn/run.py b/bsuite/baselines/jax/boot_dqn/run.py index d79de6e6..1fd05cdf 100644 --- a/bsuite/baselines/jax/boot_dqn/run.py +++ b/bsuite/baselines/jax/boot_dqn/run.py @@ -33,7 +33,7 @@ # Internal imports. -flags.DEFINE_integer('num_ensemble', 1, 'Size of ensemble.') +flags.DEFINE_integer('num_ensemble', 20, 'Size of ensemble.') # Experiment flags. flags.DEFINE_string(