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

..

data

Update collate_fn.py

2025-08-14 23:21:32 +08:00

evaluation

Improve evaluation metrics, optimize parameter naming, fix label index error

2025-08-27 20:55:39 +08:00

modeling

fix(sconet): convert label_ids to CUDA LongTensor

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

utils

enable find_unused_parameters flag for DDP

2023-11-21 19:26:42 +08:00

main.py

compatible with torch 2.0

2024-03-19 15:20:41 +08:00