Show accuracy in the progress bar
This commit is contained in:
parent
fadf8c25bf
commit
d0d69f610e
@ -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):
|
||||||
|
Loading…
Reference in New Issue
Block a user