From 73298cf24df193e7a63667a7d60bdfc4e1fa0659 Mon Sep 17 00:00:00 2001
From: Kilian Matheis <kilian.matheis@tum.de>
Date: Tue, 10 Feb 2026 16:19:34 -0700
Subject: [PATCH] Fix: Total_Loss calculation for DB-VAE

---
 lab2/solutions/PT_Part2_Debiasing_Solution.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lab2/solutions/PT_Part2_Debiasing_Solution.ipynb b/lab2/solutions/PT_Part2_Debiasing_Solution.ipynb
index 02b4351e..0cc932f0 100644
--- a/lab2/solutions/PT_Part2_Debiasing_Solution.ipynb
+++ b/lab2/solutions/PT_Part2_Debiasing_Solution.ipynb
@@ -821,7 +821,7 @@
         "\n",
         "    # TODO: define the DB-VAE total loss! Use torch.mean to average over all\n",
         "    # samples\n",
-        "    total_loss = torch.mean(classification_loss * face_indicator + vae_loss)\n",
+        "    total_loss = torch.mean(classification_loss + face_indicator * vae_loss)\n",
         "    # total_loss = # TODO\n",
         "\n",
         "    return total_loss, classification_loss"