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

[FatihcanUslu]

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)