evolutionary-policy-optimization 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

evolutionary_policy_optimization/epo.py

@@ -207,6 +207,7 @@ class LatentGenePool(Module):
         if l2norm_latent:
             latents = maybe_l2norm(latents, dim = -1)
 
+        self.num_latents = num_latents
         self.latents = nn.Parameter(latents, requires_grad = False)
 
         self.maybe_l2norm = maybe_l2norm
@@ -248,6 +249,7 @@ class LatentGenePool(Module):
         p - population
         g - gene dimension
         """
+        assert self.num_latents > 1
 
         genes = self.latents # the latents are the genes
 
@@ -309,14 +311,21 @@ class LatentGenePool(Module):
     def forward(
         self,
         *args,
-        latent_id: int,
+        latent_id: int | None = None,
         **kwargs,
     ):
 
         assert exists(self.net)
 
+        # if only 1 latent, assume doing ablation and get lone gene
+
+        if self.num_latents == 1:
+            latent_id = 0
+
         assert 0 <= latent_id < self.num_latents
 
+        # fetch latent
+
         latent = self.latents[latent_id]
 
         return self.net(

{evolutionary_policy_optimization-0.0.1.dist-info → evolutionary_policy_optimization-0.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evolutionary-policy-optimization
-Version: 0.0.1
+Version: 0.0.2
 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.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+evolutionary_policy_optimization/__init__.py,sha256=Nu-_OMA8abe4AiW9Gw6MvbZH0JZpMHMqjeXmkC9-7UU,81
+evolutionary_policy_optimization/epo.py,sha256=6IcrAooUY8csFMv4ho5bf6TVAsk1cYyXB5hG3NA-jbA,9941
+evolutionary_policy_optimization/experimental.py,sha256=ktBKxRF27Qsj7WIgBpYlWXqMVxO9zOx2oD1JuDYRAwM,548
+evolutionary_policy_optimization-0.0.2.dist-info/METADATA,sha256=XqR4EZUWWXRjskU3_eHlwNzCeEddjm20un4E04xmNLk,4072
+evolutionary_policy_optimization-0.0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+evolutionary_policy_optimization-0.0.2.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
+evolutionary_policy_optimization-0.0.2.dist-info/RECORD,,

evolutionary_policy_optimization-0.0.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-evolutionary_policy_optimization/__init__.py,sha256=Nu-_OMA8abe4AiW9Gw6MvbZH0JZpMHMqjeXmkC9-7UU,81
-evolutionary_policy_optimization/epo.py,sha256=todzxq4nLnkYhrFByP3DDeq5KpkLTxtq3Rz4VTFK59o,9698
-evolutionary_policy_optimization/experimental.py,sha256=ktBKxRF27Qsj7WIgBpYlWXqMVxO9zOx2oD1JuDYRAwM,548
-evolutionary_policy_optimization-0.0.1.dist-info/METADATA,sha256=IEhBeiiVRsY0mj--DLn2-dT0MtLg2mSGsbU24mXq0vE,4072
-evolutionary_policy_optimization-0.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-evolutionary_policy_optimization-0.0.1.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
-evolutionary_policy_optimization-0.0.1.dist-info/RECORD,,