Show accuracy in the progress bar

2021-04-21 22:28:36 +02:00 · 2021-04-21 22:28:36 +02:00 · d0d69f610e
commit d0d69f610e
parent fadf8c25bf
1 changed files with 4 additions and 2 deletions
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@ -54,10 +54,12 @@ class GLVQ(pl.LightningModule):
        self.train_acc(
            preds.int(),
            y.int())  # FloatTensors are assumed to be class probabilities
-        self.log("Training Accuracy",
+        self.log("acc",
                 self.train_acc,
                 on_step=False,
-                 on_epoch=True)
+                 on_epoch=True,
+                 prog_bar=True,
+                 logger=True)
        return loss

    # def training_epoch_end(self, outs):