From 0ee475d2dfa0a5aa5cd6da5f8895d62a8080320d Mon Sep 17 00:00:00 2001 From: lucidrains Date: Sat, 18 Oct 2025 08:50:53 -0700 Subject: [PATCH] oops --- dreamer4/dreamer4.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/dreamer4/dreamer4.py b/dreamer4/dreamer4.py index 4d4d7dc..a141007 100644 --- a/dreamer4/dreamer4.py +++ b/dreamer4/dreamer4.py @@ -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 diff --git a/pyproject.toml b/pyproject.toml index d48ab68..af698ba 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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" }