From fab786a07e1572acda5cc075433be8653d603fad Mon Sep 17 00:00:00 2001 From: Jensun Ravichandran Date: Tue, 29 Mar 2022 15:24:42 +0200 Subject: [PATCH] =?UTF-8?q?fix:=20rename=20hparam=20`output=5Fdim`=20?= =?UTF-8?q?=E2=86=92=20`latent=5Fdim`=20in=20`SiameseGMLVQ`?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/source/tutorial.ipynb | 2 +- prototorch/models/glvq.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/tutorial.ipynb b/docs/source/tutorial.ipynb index 3329848..ddadec5 100644 --- 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", diff --git a/prototorch/models/glvq.py b/prototorch/models/glvq.py index 71573b3..dc2858a 100644 --- 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, )