evolutionary-policy-optimization 0.2.1__tar.gz → 0.2.3__tar.gz

{evolutionary_policy_optimization-0.2.1 → evolutionary_policy_optimization-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.2.1
+Version: 0.2.3
 Summary: EPO - Pytorch
 Project-URL: Homepage, https://pypi.org/project/evolutionary-policy-optimization/
 Project-URL: Repository, https://github.com/lucidrains/evolutionary-policy-optimization

{evolutionary_policy_optimization-0.2.1 → evolutionary_policy_optimization-0.2.3}/evolutionary_policy_optimization/epo.py

@@ -1044,7 +1044,14 @@ class Agent(Module):
             actor.state_norm = critic.state_norm = state_norm
 
         self.use_critic_ema = use_critic_ema
-        self.critic_ema = EMA(critic, beta = critic_ema_beta, include_online_model = False, **ema_kwargs) if use_critic_ema else None
+
+        self.critic_ema = EMA(
+            critic,
+            beta = critic_ema_beta,
+            include_online_model = False,
+            ignore_startswith_names = {'state_norm'},
+            **ema_kwargs
+        ) if use_critic_ema else None
 
         self.latent_gene_pool = latent_gene_pool
         self.num_latents = latent_gene_pool.num_latents if exists(latent_gene_pool) else 1

{evolutionary_policy_optimization-0.2.1 → evolutionary_policy_optimization-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "evolutionary-policy-optimization"
-version = "0.2.1"
+version = "0.2.3"
 description = "EPO - Pytorch"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }