hongshaorou/dreamer4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
lucidrains 2025-10-05 11:52:54 -07:00
parent f507afa0d3
commit 338def693d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dreamer4/dreamer4.py
View File

@ -1036,7 +1036,7 @@ class DynamicsModel(Module):
self.action_learned_embed = Parameter(torch.randn(dim) * 1e-2) 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) self.task_embed = nn.Embedding(num_tasks, dim)
# calculate "space" seq len # calculate "space" seq len