fix: inverted logic

Former-commit-id: b3a388a527875cdc65ee3ee5c1383181bc75b6c6 [formerly 8b53736ce5adf77812831c7a3fdbca8b3b376edb] Former-commit-id: 35b5ad1b73d5f3682fdd4251e280706373d17d36
2022-07-08 13:35:08 +02:00 · 2022-07-08 13:35:08 +02:00 · 1a94fa6c09
parent 6acac70972
commit 1a94fa6c09
1 changed files with 1 additions and 1 deletions
--- a/src/unet/model.py
+++ b/src/unet/model.py
@ -195,7 +195,7 @@ class UNet(pl.LightningModule):
        accuracy = (masks_true == masks_pred_bin).float().mean()

        rows = []
-        if batch_idx % 50 == 0 or dice < 0.1:
+        if batch_idx % 50 == 0 or dice > 0.9:
            for i, (img, mask, pred, pred_bin) in enumerate(
                zip(
                    images.cpu(),