fix: rename hparam output_dim → latent_dim in SiameseGMLVQ

2022-03-29 15:24:42 +02:00
parent 40bd7ed380
commit fab786a07e
2 changed files with 2 additions and 2 deletions
--- a/prototorch/models/glvq.py
+++ b/prototorch/models/glvq.py
@@ -230,7 +230,7 @@ class SiameseGMLVQ(SiameseGLVQ):
                                       EyeTransformInitializer())
        self.backbone = LinearTransform(
            self.hparams.input_dim,
-            self.hparams.output_dim,
+            self.hparams.latent_dim,
            initializer=omega_initializer,
        )