Spaces:
Build error
Build error
PeteBleackley commited on
Commit ·
4808422
1
Parent(s): 7b60f84
Diagnostics
Browse files
qarac/models/layers/FactorizedMatrixMultiplication.py
CHANGED
|
@@ -20,4 +20,6 @@ class FactorizedMatrixMultiplication(torch.nn.Module):
|
|
| 20 |
self.matrix = torch.tensordot(self.left,self.right,1)
|
| 21 |
|
| 22 |
def forward(self,X):
|
|
|
|
|
|
|
| 23 |
return torch.einsum('ij,klj->kli',self.matrix,X)
|
|
|
|
| 20 |
self.matrix = torch.tensordot(self.left,self.right,1)
|
| 21 |
|
| 22 |
def forward(self,X):
|
| 23 |
+
print(X.device)
|
| 24 |
+
print(self.matrix.device)
|
| 25 |
return torch.einsum('ij,klj->kli',self.matrix,X)
|