From add3be0c5752faa1983f28174dddf72fc081666b Mon Sep 17 00:00:00 2001
From: Gabriel Gutierrez <gabriel.bgs00@gmail.com>
Date: Tue, 14 May 2024 09:29:45 -0300
Subject: [PATCH] log loss

---
 minerva/models/nets/setr.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/minerva/models/nets/setr.py b/minerva/models/nets/setr.py
index 109c036..a5f6ef4 100644
--- a/minerva/models/nets/setr.py
+++ b/minerva/models/nets/setr.py
@@ -604,6 +604,16 @@ def _single_step(self, batch: torch.Tensor, batch_idx: int, step_name: str):
                 sync_dist=True,
             )
 
+        self.log(
+            f"{step_name}_loss",
+            loss,
+            on_step=True,
+            on_epoch=True,
+            prog_bar=True,
+            logger=True,
+            sync_dist=True,
+        )
+
         return loss
 
     def training_step(self, batch: torch.Tensor, batch_idx: int):