fix: remove .to(device) from dataset to avoid pin_memory error in Trainer

train_abuse_model.py

@@ -41,13 +41,14 @@ class AbuseDataset(Dataset):
 
     def __len__(self):
         return len(self.labels)
-
+    
     def __getitem__(self, idx):
-        item = {key: torch.tensor(val[idx]).to(device) for key, val in self.encodings.items()}
-        item["labels"] = torch.tensor(self.labels[idx], dtype=torch.float).to(device)
+        item = {key: torch.tensor(val[idx]) for key, val in self.encodings.items()}
+        item["labels"] = torch.tensor(self.labels[idx], dtype=torch.float)
         return item
 
 
+
 #  Convert label values to soft scores: "yes" = 1.0, "plausibly" = 0.5, others = 0.0
 def label_row_soft(row):
     labels = []