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

Route initialized prototypes

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-07 21:18:08 +02:00
parent 43fc7d1678
commit 022d791ea5
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
prototorch/models/abstract.py
View File

@ -81,10 +81,12 @@ class UnsupervisedPrototypeModel(PrototypeModel):
# Layers
prototype_initializer = kwargs.get("prototype_initializer", None)
if prototype_initializer is not None:
initialized_prototypes = kwargs.get("initialized_prototypes", None)
if prototype_initializer is not None or initialized_prototypes is not None:
self.proto_layer = Components(
self.hparams.num_prototypes,
initializer=prototype_initializer,
initialized_components=initialized_prototypes,
)
def compute_distances(self, x):
@ -103,10 +105,12 @@ class SupervisedPrototypeModel(PrototypeModel):
# Layers
prototype_initializer = kwargs.get("prototype_initializer", None)
if prototype_initializer is not None:
initialized_prototypes = kwargs.get("initialized_prototypes", None)
if prototype_initializer is not None or initialized_prototypes is not None:
self.proto_layer = LabeledComponents(
distribution=self.hparams.distribution,
initializer=prototype_initializer,
initialized_components=initialized_prototypes,
)
self.competition_layer = WTAC()