evolutionary-policy-optimization 0.0.68__py3-none-any.whl → 0.0.69__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 -2
- {evolutionary_policy_optimization-0.0.68.dist-info → evolutionary_policy_optimization-0.0.69.dist-info}/METADATA +1 -1
- {evolutionary_policy_optimization-0.0.68.dist-info → evolutionary_policy_optimization-0.0.69.dist-info}/RECORD +5 -5
- {evolutionary_policy_optimization-0.0.68.dist-info → evolutionary_policy_optimization-0.0.69.dist-info}/WHEEL +0 -0
- {evolutionary_policy_optimization-0.0.68.dist-info → evolutionary_policy_optimization-0.0.69.dist-info}/licenses/LICENSE +0 -0
|
@@ -681,6 +681,8 @@ class Agent(Module):
|
|
|
681
681
|
actor_lr = 1e-4,
|
|
682
682
|
critic_lr = 1e-4,
|
|
683
683
|
latent_lr = 1e-5,
|
|
684
|
+
actor_weight_decay = 1e-3,
|
|
685
|
+
critic_weight_decay = 1e-3,
|
|
684
686
|
diversity_aux_loss_weight = 0.,
|
|
685
687
|
use_critic_ema = True,
|
|
686
688
|
critic_ema_beta = 0.99,
|
|
@@ -737,8 +739,8 @@ class Agent(Module):
|
|
|
737
739
|
|
|
738
740
|
# optimizers
|
|
739
741
|
|
|
740
|
-
self.actor_optim = optim_klass(actor.parameters(), lr = actor_lr, **actor_optim_kwargs)
|
|
741
|
-
self.critic_optim = optim_klass(critic.parameters(), lr = critic_lr, **critic_optim_kwargs)
|
|
742
|
+
self.actor_optim = optim_klass(actor.parameters(), lr = actor_lr, weight_decay = actor_weight_decay, **actor_optim_kwargs)
|
|
743
|
+
self.critic_optim = optim_klass(critic.parameters(), lr = critic_lr, weight_decay = critic_weight_decay, **critic_optim_kwargs)
|
|
742
744
|
|
|
743
745
|
self.latent_optim = optim_klass(latent_gene_pool.parameters(), lr = latent_lr, **latent_optim_kwargs) if exists(latent_gene_pool) and not latent_gene_pool.frozen_latents else None
|
|
744
746
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: evolutionary-policy-optimization
|
|
3
|
-
Version: 0.0.
|
|
3
|
+
Version: 0.0.69
|
|
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
|
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
evolutionary_policy_optimization/__init__.py,sha256=0q0aBuFgWi06MLMD8FiHzBYQ3_W4LYWrwmCtF3u5H2A,201
|
|
2
2
|
evolutionary_policy_optimization/distributed.py,sha256=7KgZdeS_wxBHo_du9XZFB1Cu318J-Bp66Xdr6Log_20,2423
|
|
3
|
-
evolutionary_policy_optimization/epo.py,sha256=
|
|
3
|
+
evolutionary_policy_optimization/epo.py,sha256=e83tghTNXfCW0zhhb4nIjvfbzDvzWRxgTlm3vKJd4rM,38189
|
|
4
4
|
evolutionary_policy_optimization/experimental.py,sha256=-IgqjJ_Wk_CMB1y9YYWpoYqTG9GZHAS6kbRdTluVevg,1563
|
|
5
5
|
evolutionary_policy_optimization/mock_env.py,sha256=Bv9ONFRbma8wpjUurc9aCk19A6ceiWitRnS3nwrIR64,1339
|
|
6
|
-
evolutionary_policy_optimization-0.0.
|
|
7
|
-
evolutionary_policy_optimization-0.0.
|
|
8
|
-
evolutionary_policy_optimization-0.0.
|
|
9
|
-
evolutionary_policy_optimization-0.0.
|
|
6
|
+
evolutionary_policy_optimization-0.0.69.dist-info/METADATA,sha256=UZEaCY5lfTRMkuyEQs5PLA1AZzSOcsRzXey9kgdd9i0,6220
|
|
7
|
+
evolutionary_policy_optimization-0.0.69.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
|
|
8
|
+
evolutionary_policy_optimization-0.0.69.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
|
|
9
|
+
evolutionary_policy_optimization-0.0.69.dist-info/RECORD,,
|
|
File without changes
|