evolutionary-policy-optimization 0.0.33__py3-none-any.whl → 0.0.35__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- evolutionary_policy_optimization/epo.py +4 -1
- {evolutionary_policy_optimization-0.0.33.dist-info → evolutionary_policy_optimization-0.0.35.dist-info}/METADATA +2 -1
- {evolutionary_policy_optimization-0.0.33.dist-info → evolutionary_policy_optimization-0.0.35.dist-info}/RECORD +5 -5
- {evolutionary_policy_optimization-0.0.33.dist-info → evolutionary_policy_optimization-0.0.35.dist-info}/WHEEL +0 -0
- {evolutionary_policy_optimization-0.0.33.dist-info → evolutionary_policy_optimization-0.0.35.dist-info}/licenses/LICENSE +0 -0
|
@@ -558,6 +558,7 @@ class LatentGenePool(Module):
|
|
|
558
558
|
state: Tensor | None = None,
|
|
559
559
|
latent_id: int | None = None,
|
|
560
560
|
net: Module | None = None,
|
|
561
|
+
net_latent_kwarg_name = 'latent',
|
|
561
562
|
**kwargs,
|
|
562
563
|
):
|
|
563
564
|
device = self.latents.device
|
|
@@ -583,9 +584,11 @@ class LatentGenePool(Module):
|
|
|
583
584
|
if not exists(net):
|
|
584
585
|
return latent
|
|
585
586
|
|
|
587
|
+
latent_kwarg = {net_latent_kwarg_name: latent}
|
|
588
|
+
|
|
586
589
|
return net(
|
|
587
590
|
*args,
|
|
588
|
-
|
|
591
|
+
**latent_kwarg,
|
|
589
592
|
**kwargs
|
|
590
593
|
)
|
|
591
594
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: evolutionary-policy-optimization
|
|
3
|
-
Version: 0.0.
|
|
3
|
+
Version: 0.0.35
|
|
4
4
|
Summary: EPO - Pytorch
|
|
5
5
|
Project-URL: Homepage, https://pypi.org/project/evolutionary-policy-optimization/
|
|
6
6
|
Project-URL: Repository, https://github.com/lucidrains/evolutionary-policy-optimization
|
|
@@ -38,6 +38,7 @@ Requires-Dist: adam-atan2-pytorch
|
|
|
38
38
|
Requires-Dist: assoc-scan
|
|
39
39
|
Requires-Dist: einops>=0.8.1
|
|
40
40
|
Requires-Dist: einx>=0.3.0
|
|
41
|
+
Requires-Dist: ema-pytorch>=0.7.7
|
|
41
42
|
Requires-Dist: hl-gauss-pytorch>=0.1.19
|
|
42
43
|
Requires-Dist: torch>=2.2
|
|
43
44
|
Requires-Dist: tqdm
|
|
@@ -1,8 +1,8 @@
|
|
|
1
1
|
evolutionary_policy_optimization/__init__.py,sha256=Qavcia0n13jjaWIS_LPW7QrxSLT_BBeKujCjF9kQjbA,133
|
|
2
|
-
evolutionary_policy_optimization/epo.py,sha256=
|
|
2
|
+
evolutionary_policy_optimization/epo.py,sha256=OVI2XCAWYtggA4TRVDmXecQkXfXB1r-HchcXGWsvIvg,23941
|
|
3
3
|
evolutionary_policy_optimization/experimental.py,sha256=ktBKxRF27Qsj7WIgBpYlWXqMVxO9zOx2oD1JuDYRAwM,548
|
|
4
4
|
evolutionary_policy_optimization/mock_env.py,sha256=3xrd-gwjZeVd_sEvxIyX0lppnMWcfQGOapO-XjKmExI,816
|
|
5
|
-
evolutionary_policy_optimization-0.0.
|
|
6
|
-
evolutionary_policy_optimization-0.0.
|
|
7
|
-
evolutionary_policy_optimization-0.0.
|
|
8
|
-
evolutionary_policy_optimization-0.0.
|
|
5
|
+
evolutionary_policy_optimization-0.0.35.dist-info/METADATA,sha256=mkHr6X8PUDj_cYBEPo4KgW9COSm70Tud_6ckgtvZ-Ds,4992
|
|
6
|
+
evolutionary_policy_optimization-0.0.35.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
7
|
+
evolutionary_policy_optimization-0.0.35.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
|
|
8
|
+
evolutionary_policy_optimization-0.0.35.dist-info/RECORD,,
|
|
File without changes
|