/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:312: operator(): block: [0,0,0], thread: [1,0,0] Assertion idx_dim >= 0 && idx_dim < index_size && "index out of bounds" failed.
/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:312: operator(): block: [0,0,0], thread: [40,0,0] Assertion idx_dim >= 0 && idx_dim < index_size && "index out of bounds" failed.
/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:312: operator(): block: [0,0,0], thread: [6,0,0] Assertion idx_dim >= 0 && idx_dim < index_size && "index out of bounds" failed.
Traceback (most recent call last):
File “main.py”, line 954, in 
main()
File “main.py”, line 665, in main
optimizers=optimizers)
File “main.py”, line 773, in train_one_epoch
label_size=args.token_label_size)
File “/opt/conda/lib/python3.6/site-packages/tlt/data/mixup.py”, line 90, in mixup_target
y1 = get_labelmaps_with_coords(target, num_classes, on_value=on_value, off_value=off_value, device=device, label_size=label_size)
File “/opt/conda/lib/python3.6/site-packages/tlt/data/mixup.py”, line 64, in get_labelmaps_with_coords
num_classes=num_classes,device=device)
File “/opt/conda/lib/python3.6/site-packages/tlt/data/mixup.py”, line 12, in get_featuremaps
label_maps_topk_sizes[3]], 0, dtype=torch.float32 ,device=device)
RuntimeError: CUDA error: device-side assert triggered
terminate called after throwing an instance of ‘std::runtime_error’
what(): NCCL error in: /pytorch/torch/lib/c10d/…/c10d/NCCLUtils.hpp:136, unhandled cuda error, NCCL version 2.7.8