julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: rename hparam output_dim → latent_dim in SiameseGMLVQ

Browse Source
This commit is contained in:
Jensun Ravichandran 2022-03-29 15:24:42 +02:00
parent 40bd7ed380
commit fab786a07e
No known key found for this signature in database
GPG Key ID: 4E9348239810B51F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/tutorial.ipynb
View File

@ -422,7 +422,7 @@
"trained_model = pt.models.GLVQ.load_from_checkpoint(\"./checkpoints/glvq_iris.ckpt\", strict=False)\n", "trained_model = pt.models.GLVQ.load_from_checkpoint(\"./checkpoints/glvq_iris.ckpt\", strict=False)\n",
"model = pt.models.SiameseGMLVQ(\n", "model = pt.models.SiameseGMLVQ(\n",
" dict(input_dim=2,\n", " dict(input_dim=2,\n",
" output_dim=2,\n", " latent_dim=2,\n",
" distribution=(3, 2),\n", " distribution=(3, 2),\n",
" proto_lr=0.0001,\n", " proto_lr=0.0001,\n",
" bb_lr=0.0001),\n", " bb_lr=0.0001),\n",

2
prototorch/models/glvq.py
View File

@ -230,7 +230,7 @@ class SiameseGMLVQ(SiameseGLVQ):
EyeTransformInitializer()) EyeTransformInitializer())
self.backbone = LinearTransform( self.backbone = LinearTransform(
self.hparams.input_dim, self.hparams.input_dim,
self.hparams.output_dim, self.hparams.latent_dim,
initializer=omega_initializer, initializer=omega_initializer,
) )