evolutionary-policy-optimization 0.2.0__py3-none-any.whl → 0.2.3__py3-none-any.whl

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
@@ -1179,8 +1186,6 @@ class Agent(Module):
 
         unwrap(self.critic).load_state_dict(pkg['critic'])
 
-        unwrap(self.latent_gene_pool).load_state_dict(pkg['latents'])
-
         if self.use_critic_ema:
             self.critic_ema.load_state_dict(pkg['critic_ema'])
 

{evolutionary_policy_optimization-0.2.0.dist-info → evolutionary_policy_optimization-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.2.0
+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.0.dist-info → evolutionary_policy_optimization-0.2.3.dist-info}/RECORD

@@ -1,10 +1,10 @@
 evolutionary_policy_optimization/__init__.py,sha256=NyiYDYU7DlpmOTM7xiBQET3r1WwX0ebrgMCBLSQrW3c,288
 evolutionary_policy_optimization/distributed.py,sha256=MxyxqxANAuOm8GYb0Yu09EHd_aVLhK2uwgrfuVWciPU,2342
 evolutionary_policy_optimization/env_wrappers.py,sha256=bDL06o9_b1iW6k3fw2xifnOnYlzs643tdW6Yv2gsIdw,803
-evolutionary_policy_optimization/epo.py,sha256=sMjCvwT6upNZ48DHegtZ40K9M8PloeW8KBCedx2fc-4,51796
+evolutionary_policy_optimization/epo.py,sha256=81rf249ykaPrAEMGk9KsF98qDkCUhW8xL3-2UXIvI2E,51838
 evolutionary_policy_optimization/experimental.py,sha256=ZyOGHbE4dXmt4zCljSzcUklua4vlOwQtslhFEm0JN94,1716
 evolutionary_policy_optimization/mock_env.py,sha256=TLyyRm6tOD0Kdn9QqJJQriaSnsR-YmNQHo4OohmZFG4,1410
-evolutionary_policy_optimization-0.2.0.dist-info/METADATA,sha256=mqgk3bcYYLwGYYoOqWclUV2QuitxC_shw8nc6hrP8K0,8697
-evolutionary_policy_optimization-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-evolutionary_policy_optimization-0.2.0.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
-evolutionary_policy_optimization-0.2.0.dist-info/RECORD,,
+evolutionary_policy_optimization-0.2.3.dist-info/METADATA,sha256=syjis1-9dDCEwfGt7CeMfhC5k7OegIl4BBsKaTnYssQ,8697
+evolutionary_policy_optimization-0.2.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+evolutionary_policy_optimization-0.2.3.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
+evolutionary_policy_optimization-0.2.3.dist-info/RECORD,,