dreamer4 0.0.97__tar.gz → 0.0.99__tar.gz

{dreamer4-0.0.97 → dreamer4-0.0.99}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dreamer4
-Version: 0.0.97
+Version: 0.0.99
 Summary: Dreamer 4
 Project-URL: Homepage, https://pypi.org/project/dreamer4/
 Project-URL: Repository, https://github.com/lucidrains/dreamer4

{dreamer4-0.0.97 → dreamer4-0.0.99}/dreamer4/dreamer4.py

@@ -2580,6 +2580,8 @@ class DynamicsWorldModel(Module):
             if self.pmpo_kl_div_loss_weight > 0.:
                 new_unembedded_actions = self.action_embedder.unembed(policy_embed, pred_head_index = 0)
 
+                # mentioned that the "reverse direction for the prior KL" was used
+
                 discrete_kl_div, continuous_kl_div = self.action_embedder.kl_div(old_action_unembeds, new_unembedded_actions)
 
                 # accumulate discrete and continuous kl div
@@ -2868,7 +2870,7 @@ class DynamicsWorldModel(Module):
                 # maybe store old actions
 
                 if store_old_action_unembeds:
-                    acc_policy_embed = safe_cat((acc_policy_embed, policy_embed))
+                    acc_policy_embed = safe_cat((acc_policy_embed, policy_embed), dim = 1)
 
                 # sample actions
 

{dreamer4-0.0.97 → dreamer4-0.0.99}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "dreamer4"
-version = "0.0.97"
+version = "0.0.99"
 description = "Dreamer 4"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }