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

fix: forward of LinearTransform uses undetached weights now

Browse Source
This commit is contained in:
Alexander Engelsberger
2022-03-29 17:06:57 +02:00
parent ed5b9b6c62
commit 46ff1c4eb1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
prototorch/core/transforms.py
View File

@@ -36,7 +36,7 @@ class LinearTransform(torch.nn.Module):
self._register_weights(weights) self._register_weights(weights)
def forward(self, x): def forward(self, x):
return x @ self.weights return x @ self._weights
# Aliases # Aliases