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

feat: add omega parameter api

Browse Source
This commit is contained in:
Alexander Engelsberger 2022-08-19 12:15:11 +02:00
parent e0abb1f3de
commit ec294bdd37
No known key found for this signature in database
GPG Key ID: DE8669706B6AC2E7
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
prototorch/models/architectures/comparison.py
View File

@ -137,3 +137,12 @@ class OmegaComparisonMixin(SimpleComparisonMixin):
''' '''
lam = self.lambda_matrix lam = self.lambda_matrix
return lam.abs().sum(0) return lam.abs().sum(0)
@property
def parameter_omega(self):
return self._omega
@parameter_omega.setter
def parameter_omega(self, new_omega):
with torch.no_grad():
self._omega.data.copy_(new_omega)