dreamer4 0.0.87__py3-none-any.whl → 0.0.88__py3-none-any.whl

dreamer4/dreamer4.py

@@ -2255,7 +2255,7 @@ class DynamicsWorldModel(Module):
             video = cat((video, next_frame), dim = 2)
             rewards = safe_cat((rewards, reward), dim = 1)
 
-            acc_agent_embed = safe_cat((acc_agent_embed, agent_embed), dim = 1)
+            acc_agent_embed = safe_cat((acc_agent_embed, one_agent_embed), dim = 1)
 
         # package up one experience for learning
 
@@ -2397,7 +2397,7 @@ class DynamicsWorldModel(Module):
                     return_intermediates = True
                 )
 
-        agent_embeds = agent_embeds[..., agent_index, :]
+            agent_embeds = agent_embeds[..., agent_index, :]
 
         # maybe detach agent embed
 
@@ -2672,7 +2672,9 @@ class DynamicsWorldModel(Module):
 
             # maybe store agent embed
 
-            acc_agent_embed = safe_cat((acc_agent_embed, agent_embed), dim = 1)
+            if store_agent_embed:
+                one_agent_embed = agent_embed[:, -1:, agent_index]
+                acc_agent_embed = safe_cat((acc_agent_embed, one_agent_embed), dim = 1)
 
             # decode the agent actions if needed
 

{dreamer4-0.0.87.dist-info → dreamer4-0.0.88.dist-info}/METADATA

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

dreamer4-0.0.88.dist-info/RECORD

@@ -0,0 +1,8 @@
+dreamer4/__init__.py,sha256=Jssh1obzDRtTfBLZl36kXge1cIQlMjf_8DyjPulvKSk,183
+dreamer4/dreamer4.py,sha256=qAVLInnB5kgazzyp1KFIgWEIjJyLhlUQb3RNmybi23g,112219
+dreamer4/mocks.py,sha256=TfqOB_Gq6N_GggBYwa6ZAJQx38ntlYbXZe23Ne4jshw,2502
+dreamer4/trainers.py,sha256=D2b7WTgTHElLhIWLFgl2Ct2knGJLTk91HHpC5UkNvG0,14028
+dreamer4-0.0.88.dist-info/METADATA,sha256=cI2FsEuCzaLcqobMAtLkN_6nBv6iDZQNYr1hlV7YmlY,3065
+dreamer4-0.0.88.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+dreamer4-0.0.88.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
+dreamer4-0.0.88.dist-info/RECORD,,

dreamer4-0.0.87.dist-info/RECORD

@@ -1,8 +0,0 @@
-dreamer4/__init__.py,sha256=Jssh1obzDRtTfBLZl36kXge1cIQlMjf_8DyjPulvKSk,183
-dreamer4/dreamer4.py,sha256=CF_nzckZT-WX_N2FO1xLXb4VjillI-MVw2poiuTVEck,112102
-dreamer4/mocks.py,sha256=TfqOB_Gq6N_GggBYwa6ZAJQx38ntlYbXZe23Ne4jshw,2502
-dreamer4/trainers.py,sha256=D2b7WTgTHElLhIWLFgl2Ct2knGJLTk91HHpC5UkNvG0,14028
-dreamer4-0.0.87.dist-info/METADATA,sha256=Gu4RFEmdDUz1CtBLSIr9MJLC1GXwtE34qTXe2ntpmUc,3065
-dreamer4-0.0.87.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-dreamer4-0.0.87.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
-dreamer4-0.0.87.dist-info/RECORD,,