fix: rename hparam output_dim → latent_dim in SiameseGMLVQ

2022-03-29 15:24:42 +02:00 · 2022-03-29 15:24:42 +02:00 · fab786a07e
commit fab786a07e
parent 40bd7ed380
2 changed files with 2 additions and 2 deletions
--- a/docs/source/tutorial.ipynb
+++ b/docs/source/tutorial.ipynb
@ -422,7 +422,7 @@
    "trained_model = pt.models.GLVQ.load_from_checkpoint(\"./checkpoints/glvq_iris.ckpt\", strict=False)\n",
    "model = pt.models.SiameseGMLVQ(\n",
    "    dict(input_dim=2,\n",
-    "         output_dim=2,\n",
+    "         latent_dim=2,\n",
    "         distribution=(3, 2),\n",
    "         proto_lr=0.0001,\n",
    "         bb_lr=0.0001),\n",
--- 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,
        )