Show accuracy in the progress bar

This commit is contained in:
Jensun Ravichandran 2021-04-21 22:28:36 +02:00
parent fadf8c25bf
commit d0d69f610e

View File

@ -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):