Diagnostics

qarac/models/QaracTrainerModel.py

@@ -98,4 +98,9 @@ class QaracTrainerModel(torch.nn.Module):
         s1 = self.answer_encoder(statement1)
         
         consistency = self.cosine(s0,s1)
-        return (encode_decode,question_answering,reasoning,consistency)
+        return (encode_decode,question_answering,reasoning,consistency)
+    
+    def devices(self):
+        print("Question encoder",self.question_encoder.device)
+        print('Answer Encoder',self.answer.encoder.device)
+        print('Decoer',self.decoder.device)

qarac/models/layers/FactorizedMatrixMultiplication.py

@@ -20,6 +20,4 @@ class FactorizedMatrixMultiplication(torch.nn.Module):
         self.matrix = torch.tensordot(self.left,self.right,1)
         
     def forward(self,X):
-        print(X.device)
-        print(self.matrix.device)
         return torch.einsum('ij,klj->kli',self.matrix,X)