From 3ace8d2cc47853a1e394e6220030fea8a3c077e0 Mon Sep 17 00:00:00 2001
From: Gabriel Gutierrez <gabriel.bgs00@gmail.com>
Date: Tue, 14 May 2024 09:24:46 -0300
Subject: [PATCH] log fix setr

---
 minerva/models/nets/setr.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/minerva/models/nets/setr.py b/minerva/models/nets/setr.py
index 7a91827..4d83130 100644
--- a/minerva/models/nets/setr.py
+++ b/minerva/models/nets/setr.py
@@ -594,8 +594,9 @@ def _single_step(self, batch: torch.Tensor, batch_idx: int, step_name: str):
 
         metrics = self._compute_metrics(y_hat[0], y, step_name)
         for metric_name, metric_value in metrics.items():
-            self.log_dict(
-                {metric_name: metric_value},
+            self.log(
+                metric_name,
+                metric_value,
                 on_step=False,
                 on_epoch=True,
                 prog_bar=True,