diff --git a/train.py b/train.py index 4342575..753e1be 100644 --- a/train.py +++ b/train.py @@ -119,7 +119,7 @@ def training(dataset, opt, pipe, testing_iterations, saving_iterations, checkpoi alpha_mask = viewpoint_cam.alpha_mask.cuda() image *= alpha_mask - # Loss + # Loss test gt_image = viewpoint_cam.original_image.cuda() Ll1 = l1_loss(image, gt_image) if FUSED_SSIM_AVAILABLE: