oops

2025-10-05 11:52:54 -07:00 · 2025-10-05 11:52:54 -07:00 · 338def693d
commit 338def693d
parent f507afa0d3
1 changed files with 1 additions and 1 deletions
--- a/dreamer4/dreamer4.py
+++ b/dreamer4/dreamer4.py
@ -1036,7 +1036,7 @@ class DynamicsModel(Module):

        self.action_learned_embed = Parameter(torch.randn(dim) * 1e-2)

-        self.num_task = num_tasks
+        self.num_tasks = num_tasks
        self.task_embed = nn.Embedding(num_tasks, dim)

        # calculate "space" seq len