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

Fix bug in visualization callback

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-04-27 12:49:04 +02:00
parent 3148684812
commit 8d57f69c9e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/glvq_iris_v1.py
View File

@ -27,6 +27,7 @@ class VisualizationCallback(pl.Callback):
def on_epoch_end(self, trainer, pl_module): def on_epoch_end(self, trainer, pl_module):
protos = pl_module.prototypes protos = pl_module.prototypes
plabels = pl_module.prototype_labels plabels = pl_module.prototype_labels
x_train, y_train = self.x_train, self.y_train
ax = self.fig.gca() ax = self.fig.gca()
ax.cla() ax.cla()
ax.set_title(self.title) ax.set_title(self.title)