Fix loss

htem · Nov 7, 2023 · 15db843 · 15db843
1 parent 474cfe4
commit 15db843
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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(
             {