ML UnitTest

Knox-AAU · Dec 11, 2023 · d16b7be · d16b7be
1 parent 471c7cf
commit d16b7be
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/concept_linking/solutions/MachineLearning/src/model_training.py b/concept_linking/solutions/MachineLearning/src/model_training.py
@@ -11,8 +11,11 @@
 
 def custom_collate(batch):
     # Separate inputs, targets, and lengths
+    #inputs = [torch.LongTensor(item['input']) for item in batch]
+    #targets = [torch.tensor(item['target'], dtype=torch.float32) for item in batch]
+    #lengths = [item['length'] for item in batch]
     inputs = [torch.LongTensor(item['input']) for item in batch]
-    targets = [torch.tensor(item['target'], dtype=torch.float32) for item in batch]
+    targets = [item['target'].clone().detach() for item in batch]
     lengths = [item['length'] for item in batch]
 
     # Pad sequences