prg_bar

Author: bartzbeielstein

pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "spotPython"
-version = "0.2.44"
+version = "0.2.45"
 authors = [
   { name="T. Bartz-Beielstein", email="tbb@bartzundbartz.de" }
 ]

src/spotPython/light/csvmodel.py

@@ -72,7 +72,7 @@ def validation_step(self, batch, batch_idx, prog_bar=False):
         preds = torch.argmax(logits, dim=1)
         acc = accuracy(preds, y, task="multiclass", num_classes=self._L_out)
         self.valid_mapk(logits, y)
-        self.log("valid_mapk", self.valid_mapk, on_step=False, on_epoch=True)
+        self.log("valid_mapk", self.valid_mapk, on_step=False, on_epoch=True, prog_bar=prog_bar)
         self.log("val_loss", loss, prog_bar=prog_bar)
         self.log("val_acc", acc, prog_bar=prog_bar)
 
@@ -84,7 +84,7 @@ def test_step(self, batch, batch_idx, prog_bar=False):
         preds = torch.argmax(logits, dim=1)
         acc = accuracy(preds, y, task="multiclass", num_classes=self._L_out)
         self.test_mapk(logits, y)
-        self.log("test_mapk", self.test_mapk, on_step=True, on_epoch=True)
+        self.log("test_mapk", self.test_mapk, on_step=True, on_epoch=True, prog_bar=prog_bar)
         self.log("val_loss", loss, prog_bar=prog_bar)
         self.log("val_acc", acc, prog_bar=prog_bar)
         return loss, acc