From 1a94fa6c09dab62191b9b2743f03ec62fafc1e6f Mon Sep 17 00:00:00 2001
From: Laurent Fainsin <laurentfainsin@protonmail.com>
Date: Fri, 8 Jul 2022 13:35:08 +0200
Subject: [PATCH] fix: inverted logic

Former-commit-id: b3a388a527875cdc65ee3ee5c1383181bc75b6c6 [formerly 8b53736ce5adf77812831c7a3fdbca8b3b376edb]
Former-commit-id: 35b5ad1b73d5f3682fdd4251e280706373d17d36
---
 src/unet/model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/unet/model.py b/src/unet/model.py
index 93872f5..4556640 100644
--- 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(),