From 2b99584a27832bad383a20a5c794c9aacf680ff7 Mon Sep 17 00:00:00 2001 From: templierpaul Date: Fri, 24 Nov 2023 17:27:48 +0100 Subject: [PATCH] Fixed issue in RND emitter with es_params not passed to ES --- qdax/core/emitters/cma_me_policies.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qdax/core/emitters/cma_me_policies.py b/qdax/core/emitters/cma_me_policies.py index 96ba4504..dae99efd 100644 --- a/qdax/core/emitters/cma_me_policies.py +++ b/qdax/core/emitters/cma_me_policies.py @@ -343,7 +343,7 @@ def init( self.es = Strategies[self.es_type]( num_dims=self.reshaper.genotype_dim, popsize=self.batch_size, - sigma_init=self._sigma_g, + **self.es_params, ) # Initialize repertoire with default values