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

Return prototypes as torch tensor

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-05-07 15:45:37 +02:00
parent 63a5a98491
commit 11b3e53ecb
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/models/abstract.py
View File

@ -20,4 +20,4 @@ class AbstractLightningModel(pl.LightningModule):
class AbstractPrototypeModel(AbstractLightningModel):
@property
def prototypes(self):
return self.proto_layer.components.detach().numpy()
return self.proto_layer.components.detach().cpu()