Zzier 28d4edc647 fix(sconet): convert label_ids to CUDA LongTensor ...

- ScoNet:
  - Previously, `label_ids` remained a NumPy array, which could cause dtype/device mismatches when used with PyTorch tensors on GPU.
  - Convert `label_ids` to `torch.from_numpy(...).cuda().long()` to ensure correct tensor type (Long) and device (CUDA), aligning with loss functions that expect class indices on the same device.

2025-10-28 17:45:00 +08:00

1.9 KiB

Raw Permalink Blame History

View Raw