Add the namespace hook for GMLVQ in the model class

This commit is contained in:
Jensun Ravichandran 2021-05-09 20:53:31 +02:00
parent ff7a1e93d2
commit ca4c9da10a

View File

@ -191,6 +191,9 @@ class GMLVQ(GLVQ):
self.hparams.latent_dim,
bias=False)
# Namespace hook for the visualization callbacks to work
self.backbone = self.omega_layer
@property
def omega_matrix(self):
return self.omega_layer.weight.detach().cpu()