fix: rename hparam output_dim
→ latent_dim
in SiameseGMLVQ
This commit is contained in:
parent
40bd7ed380
commit
fab786a07e
@ -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",
|
||||
|
@ -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,
|
||||
)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user