typo in min cost selection

mittagessen · Sep 30, 2024 · 3561357 · 3561357
1 parent 6056146
commit 3561357
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kraken/lib/train/segmentation.py b/kraken/lib/train/segmentation.py
@@ -279,7 +279,7 @@ def validation_step(self, batch, batch_idx):
                 if pred_curves:
                     pred_curves = torch.stack(pred_curves)
                     cost_curves = torch.cdist(pred_curves, target_curves, p=1).cpu()
-                    costs = cost_curves = [linear_sum_assignment(cost_curves)]
+                    costs = cost_curves[linear_sum_assignment(cost_curves)]
                     # num of predictions differs from target -> take n best
                     # predictions and add error penalty term for the rest.
                     if diff := abs(len(pred_curves) - len(target_curves)):