fix: forward of LinearTransform uses undetached weights now

2022-03-29 17:06:57 +02:00 · 2022-03-29 17:06:57 +02:00 · 46ff1c4eb1
commit 46ff1c4eb1
parent ed5b9b6c62
1 changed files with 1 additions and 1 deletions
--- a/prototorch/core/transforms.py
+++ b/prototorch/core/transforms.py
@ -36,7 +36,7 @@ class LinearTransform(torch.nn.Module):
        self._register_weights(weights)

    def forward(self, x):
-        return x @ self.weights
+        return x @ self._weights


 # Aliases