evolutionary-policy-optimization 0.0.52__py3-none-any.whl → 0.0.53__py3-none-any.whl

evolutionary_policy_optimization/epo.py

@@ -360,10 +360,11 @@ class LatentGenePool(Module):
         frac_natural_selected = 0.25,    # number of least fit genes to remove from the pool
         frac_elitism = 0.1,              # frac of population to preserve from being noised
         frac_migrate = 0.1,              # frac of population, excluding elites, that migrate between islands randomly. will use a designated set migration pattern (since for some reason using random it seems to be worse for me)
-        migrate_every = 100,             # how many steps before a migration between islands
         mutation_strength = 1.,          # factor to multiply to gaussian noise as mutation to latents
         should_run_genetic_algorithm: Module | None = None, # eq (3) in paper
         default_should_run_ga_gamma = 1.5,
+        migrate_every = 100,                 # how many steps before a migration between islands
+        apply_genetic_algorithm_every = 2    # how many steps before crossover + mutation happens for genes
     ):
         super().__init__()
 
@@ -413,7 +414,10 @@ class LatentGenePool(Module):
         self.should_run_genetic_algorithm = should_run_genetic_algorithm
 
         self.can_migrate = num_islands > 1
+
         self.migrate_every = migrate_every
+        self.apply_genetic_algorithm_every = apply_genetic_algorithm_every
+
         self.register_buffer('step', tensor(1))
 
     def get_distance(self):
@@ -483,6 +487,10 @@ class LatentGenePool(Module):
     ):
         device = self.latents.device
 
+        if not divisible_by(self.step.item(), self.apply_genetic_algorithm_every):
+            self.advance_step_()
+            return
+
         """
         i - islands
         p - population

{evolutionary_policy_optimization-0.0.52.dist-info → evolutionary_policy_optimization-0.0.53.dist-info}/METADATA

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

@@ -1,8 +1,8 @@
 evolutionary_policy_optimization/__init__.py,sha256=0q0aBuFgWi06MLMD8FiHzBYQ3_W4LYWrwmCtF3u5H2A,201
-evolutionary_policy_optimization/epo.py,sha256=7iGmye36JsQ1jUm_UCCrSdvC67u70EPTAp3nCbKGjO8,33675
+evolutionary_policy_optimization/epo.py,sha256=bPg-dC7rvsiNyaYY_1lrzENjq7qezi0p_vuQe5CL2a4,34000
 evolutionary_policy_optimization/experimental.py,sha256=9FrJGviLESlYysHI3i83efT9g2ZB9ha4u3K9HXN98_w,1100
 evolutionary_policy_optimization/mock_env.py,sha256=202KJ5g57wQvOzhGYzgHfBa7Y2do5uuDvl5kFg5o73g,934
-evolutionary_policy_optimization-0.0.52.dist-info/METADATA,sha256=dET2Z11ktfJYei-xFfv7SjGKTglco6dDvELIzKrLsAQ,6213
-evolutionary_policy_optimization-0.0.52.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-evolutionary_policy_optimization-0.0.52.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
-evolutionary_policy_optimization-0.0.52.dist-info/RECORD,,
+evolutionary_policy_optimization-0.0.53.dist-info/METADATA,sha256=O5dS6Cx_kgkim6JfBTH-xZfN4smWFab2aLzXKx-b8rA,6213
+evolutionary_policy_optimization-0.0.53.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+evolutionary_policy_optimization-0.0.53.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
+evolutionary_policy_optimization-0.0.53.dist-info/RECORD,,