From 3ce0b41ee2800e5712b0210636f4d3a46d1b8441 Mon Sep 17 00:00:00 2001
From: ingra14m <1261238025@qq.com>
Date: Wed, 2 Aug 2023 17:16:31 +0800
Subject: [PATCH] Add depth loss

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index cb0f405f0..cb89a7d66 100644
--- a/train.py
+++ b/train.py
@@ -82,7 +82,7 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations):
         Ll1 = l1_loss(image, gt_image)
         loss = (1.0 - opt.lambda_dssim) * Ll1 + opt.lambda_dssim * (1.0 - ssim(image, gt_image))
         depth_loss = l1_loss(depth, gt_depth) * 0.1
-        # loss = loss + depth_loss
+        loss = loss + depth_loss
         loss.backward()
 
         iter_end.record()