From 15db843dcb12c9c597b0628a8319e63ff83915b6 Mon Sep 17 00:00:00 2001 From: brianreicher Date: Tue, 7 Nov 2023 11:54:15 -0500 Subject: [PATCH] Fix loss --- src/raygun/torch/losses/WeightedMSELoss_ACLSD.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py b/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py index 7d0c6df..de3b238 100644 --- a/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py +++ b/src/raygun/torch/losses/WeightedMSELoss_ACLSD.py @@ -68,7 +68,6 @@ def forward( gt_affs=None, affs_weights=None, pred_affs_ac=None, - # affs_weights_ac=None, TODO ): self.data_dict.update( {