Port to Pytorch 2.1

bmdanielsson · bmdanielsson · commit e147e5231b00 · 2024-01-28T17:52:10.000+01:00
diff --git a/nnue_dataset.py b/nnue_dataset.py
@@ -54,8 +54,8 @@ def get_tensors(self, device):
         them = 1.0 - us
         outcome = torch.from_numpy(np.ctypeslib.as_array(self.outcome, shape=(self.size, 1))).pin_memory().to(device=device, non_blocking=True)
         score = torch.from_numpy(np.ctypeslib.as_array(self.score, shape=(self.size, 1))).pin_memory().to(device=device, non_blocking=True)
-        white = torch._sparse_coo_tensor_unsafe(iw, white_values, (self.size, self.num_inputs))
-        black = torch._sparse_coo_tensor_unsafe(ib, black_values, (self.size, self.num_inputs))
+        white = torch.sparse_coo_tensor(iw, white_values, (self.size, self.num_inputs), check_invariants=False)
+        black = torch.sparse_coo_tensor(ib, black_values, (self.size, self.num_inputs), check_invariants=False)
         white._coalesced_(True)
         black._coalesced_(True)
         return us, them, white, black, outcome, score
