Add the namespace hook for GMLVQ in the model class
This commit is contained in:
parent
ff7a1e93d2
commit
ca4c9da10a
@ -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()
|
||||
|
Loading…
Reference in New Issue
Block a user