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

[BUGFIX] Probabilistic Models work on GPU now

Browse Source
This commit is contained in:
Alexander Engelsberger 2021-06-03 14:05:44 +02:00
parent 459f7c24be
commit 1b09b1d57b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
prototorch/models/probabilistic.py
View File

@ -36,7 +36,8 @@ class ProbabilisticLVQ(GLVQ):
distances = self._forward(x)
conditional = self.conditional_distribution(distances,
self.hparams.variance)
prior = (1. / self.num_prototypes) * torch.ones(self.num_prototypes)
prior = (1. / self.num_prototypes) * torch.ones(self.num_prototypes,
device=self.device)
posterior = conditional * prior
plabels = self.proto_layer._labels
y_pred = stratified_sum(posterior, plabels)