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-18 08:50:53 -07:00
parent 8c88a33d3b
commit 0ee475d2df
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dreamer4/dreamer4.py
View File

@ -2431,10 +2431,10 @@ class DynamicsWorldModel(Module):
)
if exists(discrete_log_probs):
behavior_clone_loss = behavior_clone_loss + discrete_log_probs.sum(dim = -1).mean()
behavior_clone_loss = behavior_clone_loss - discrete_log_probs.sum(dim = -1).mean()
if exists(continuous_log_probs):
behavior_clone_loss = behavior_clone_loss + continuous_log_probs.sum(dim = -1).mean()
behavior_clone_loss = behavior_clone_loss - continuous_log_probs.sum(dim = -1).mean()
# gather losses

2
pyproject.toml
View File

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