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

Make prototype_labels non-trainable Parameters

Browse Source
This commit is contained in:
blackfly 2020-04-27 13:39:27 +02:00
parent 0cfbc0473b
commit 8a4a596035
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/modules/prototypes.py
View File

@ -162,4 +162,5 @@ class Prototypes1D(_Prototypes):
# Register module parameters # Register module parameters
self.prototypes = torch.nn.Parameter(prototypes) self.prototypes = torch.nn.Parameter(prototypes)
self.prototype_labels = prototype_labels self.prototype_labels = torch.nn.Parameter(
prototype_labels.type(dtype)).requires_grad_(False)