From ca4c9da10a85968110aa8e29736913fbc953a7b2 Mon Sep 17 00:00:00 2001
From: Jensun Ravichandran <jjensun@gmail.com>
Date: Sun, 9 May 2021 20:53:31 +0200
Subject: [PATCH] Add the namespace hook for GMLVQ in the model class

---
 prototorch/models/glvq.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/prototorch/models/glvq.py b/prototorch/models/glvq.py
index 6fe76d3..60292b6 100644
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@@ -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()