From 13064521522e60be05846475db41602b2a44d74e Mon Sep 17 00:00:00 2001
From: Laurent Fainsin <laurentfainsin@protonmail.com>
Date: Tue, 5 Jul 2022 22:52:34 +0200
Subject: [PATCH] fix: bad logging

Former-commit-id: 221ec6b6bfcf4e2e616a1db688ef8e93a2bb5bfc [formerly 0e7792974319cbc693fdae3597d44f9c4c196b4d]
Former-commit-id: 70f179df977715a747b981f7e2784e3c7ed88028
---
 src/unet/model.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/unet/model.py b/src/unet/model.py
index be5712e..008d784 100644
--- a/src/unet/model.py
+++ b/src/unet/model.py
@@ -83,7 +83,10 @@ class UNet(pl.LightningModule):
                 ),
             )
 
-        wandb.log({log_key: table})  # replace by self.log
+        wandb.log(
+            {log_key: table},
+            commit=False,
+        )  # replace by self.log
 
     def training_step(self, batch, batch_idx):
         # unpacking
@@ -153,7 +156,7 @@ class UNet(pl.LightningModule):
         mae = torch.stack([d["mae"] for d in validation_outputs]).mean()
 
         # logging
-        wandb.log(
+        self.log_dict(
             {
                 "val/accuracy": accuracy,
                 "val/bce": loss,