New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

ScatterGatherKernel.cu error rises while trying to run run_class_finetuning.py #131

Open

FatihcanUslu opened this issue Dec 18, 2024 · 0 comments

FatihcanUslu commented Dec 18, 2024

Hi i encountered an error on mixup.py while trying to run run_class_finetuning.py

While trying to run run_class_finetuning.py i get the following error:

/opt/conda/conda-bld/pytorch_1659484810403/work/aten/src/aTen/native/cuda/ScatterGatherKernel.cu:365: operator(): block: [0,0,0], thread: [31,0,0] Assertion "idx_dim >= 0 && idx_dim< index_size && "index out of bounds " failed.
Aborted (cuda dumped)

Error happens on scatter_ function. Specifically on :

mixup.py
return torch.full((x.size()[0], num_classes), off_value, device=device).scatter_(1, x, on_value)

The text was updated successfully, but these errors were encountered:

# for free to join this conversation on GitHub. Already have an account? # to comment