Upload s23dr_2026/train.py with huggingface_hub

s23dr_2026/train.py

@@ -27,7 +27,7 @@ def build_targets(data):
     targets = []
     for labels, coords, edges in zip(data["labels"], data["coords"], data["edges"]):
         edge_coords = torch.cat([coords[edges[:, 0]], coords[edges[:, 1]]], dim=-1)
-        targets.append({"labels": labels, "coords": edge_coords, "vertex_ids": edges})
+        targets.append({"labels": labels, "coords": edge_coords})
     return targets