julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Show accuracy in the progress bar

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-04-21 22:28:36 +02:00
parent fadf8c25bf
commit d0d69f610e
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
prototorch/models/glvq.py
View File

@ -54,10 +54,12 @@ class GLVQ(pl.LightningModule):
self.train_acc( self.train_acc(
preds.int(), preds.int(),
y.int()) # FloatTensors are assumed to be class probabilities y.int()) # FloatTensors are assumed to be class probabilities
self.log("Training Accuracy", self.log("acc",
self.train_acc, self.train_acc,
on_step=False, on_step=False,
on_epoch=True) on_epoch=True,
prog_bar=True,
logger=True)
return loss return loss
# def training_epoch_end(self, outs): # def training_epoch_end(self, outs):