dreamer4 0.0.72__py3-none-any.whl → 0.0.73__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.

Potentially problematic release.

This version of dreamer4 might be problematic. Click here for more details.

Files changed (6) hide show
  1. dreamer4/dreamer4.py +5 -0
  2. {dreamer4-0.0.72.dist-info → dreamer4-0.0.73.dist-info}/METADATA +1 -1
  3. dreamer4-0.0.73.dist-info/RECORD +8 -0
  4. dreamer4-0.0.72.dist-info/RECORD +0 -8
  5. {dreamer4-0.0.72.dist-info → dreamer4-0.0.73.dist-info}/WHEEL +0 -0
  6. {dreamer4-0.0.72.dist-info → dreamer4-0.0.73.dist-info}/licenses/LICENSE +0 -0
dreamer4/dreamer4.py CHANGED
@@ -82,6 +82,7 @@ class Experience:
82
82
  log_probs: tuple[Tensor, Tensor] | None = None
83
83
  values: Tensor | None = None
84
84
  step_size: int | None = None
85
+ lens: Tensor | None = None,
85
86
  agent_index: int = 0
86
87
  is_from_world_model: bool = True
87
88
 
@@ -2560,12 +2561,16 @@ class DynamicsWorldModel(Module):
2560
2561
 
2561
2562
  # returning agent actions, rewards, and log probs + values for policy optimization
2562
2563
 
2564
+ batch, device = latents.shape[0], latents.device
2565
+ experience_lens = torch.full((batch,), time_steps, device = device)
2566
+
2563
2567
  gen = Experience(
2564
2568
  latents = latents,
2565
2569
  video = video,
2566
2570
  proprio = proprio if has_proprio else None,
2567
2571
  step_size = step_size,
2568
2572
  agent_index = agent_index,
2573
+ lens = experience_lens,
2569
2574
  is_from_world_model = True
2570
2575
  )
2571
2576
 
{dreamer4-0.0.72.dist-info → dreamer4-0.0.73.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dreamer4
3
- Version: 0.0.72
3
+ Version: 0.0.73
4
4
  Summary: Dreamer 4
5
5
  Project-URL: Homepage, https://pypi.org/project/dreamer4/
6
6
  Project-URL: Repository, https://github.com/lucidrains/dreamer4
dreamer4-0.0.73.dist-info/RECORD ADDED
@@ -0,0 +1,8 @@
1
+ dreamer4/__init__.py,sha256=Jssh1obzDRtTfBLZl36kXge1cIQlMjf_8DyjPulvKSk,183
2
+ dreamer4/dreamer4.py,sha256=z9zYb9GLRGqfiXz0I6PaE94ZFFOEIy3TwmcUUS_5KvE,104842
3
+ dreamer4/mocks.py,sha256=Oi91Yv1oK0E-Wz-KDkf79xoyWzIXCvMLCr0WYCpJDLA,1482
4
+ dreamer4/trainers.py,sha256=898ye9Y1mqxGZnU_gfQS6pECibZwwyA43sL7wK_JHAU,13993
5
+ dreamer4-0.0.73.dist-info/METADATA,sha256=TwPW4CmYD__Ecv9qR8_uBlj09wmhjspkdPDMvVBdQO4,3065
6
+ dreamer4-0.0.73.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
7
+ dreamer4-0.0.73.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
8
+ dreamer4-0.0.73.dist-info/RECORD,,
dreamer4-0.0.72.dist-info/RECORD DELETED
@@ -1,8 +0,0 @@
1
- dreamer4/__init__.py,sha256=Jssh1obzDRtTfBLZl36kXge1cIQlMjf_8DyjPulvKSk,183
2
- dreamer4/dreamer4.py,sha256=o9f0Y-BIKAm_qpqxJOkAnkP_WmzNqPq92Nu5tep2tm0,104640
3
- dreamer4/mocks.py,sha256=Oi91Yv1oK0E-Wz-KDkf79xoyWzIXCvMLCr0WYCpJDLA,1482
4
- dreamer4/trainers.py,sha256=898ye9Y1mqxGZnU_gfQS6pECibZwwyA43sL7wK_JHAU,13993
5
- dreamer4-0.0.72.dist-info/METADATA,sha256=WCVEjMXAe8awAYkuBhdvyHfWugqbdsRR3WCSBQXFs18,3065
6
- dreamer4-0.0.72.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
7
- dreamer4-0.0.72.dist-info/licenses/LICENSE,sha256=1yCiA9b5nhslTavxPjsQAO-wpOnwJR9-l8LTVi7GJuk,1066
8
- dreamer4-0.0.72.dist-info/RECORD,,