fix: forward of LinearTransform uses undetached weights now
This commit is contained in:
parent
ed5b9b6c62
commit
46ff1c4eb1
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user