AnswerDotAI / AnswerDotAI/fsdp_qlora
/opt/conda/conda-bld/pytorch_1708025847130/work/aten/src/ATen/native/cuda/Loss.cu:250: nll_loss_forward_reduce_cuda_kernel_2d: block: [0,0,0], thread: [14,0,0] Assertion `t >= 0 && t < n_classes` failed.
Open
- Dominant language
- Jupyter Notebook
- Stars
- 1.6k
- Forks
- 201
- PR merge metrics
- No merged PRs in 30d
Description
when I tried to train some 'qna' style dataset like knowrohit07/know_sql get this error.
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reproducing the training failure with the knowrohit07/know_sql dataset and inspect the reported PyTorch CUDA assertion at aten/src/ATen/native/cuda/Loss.cu:250. Compare the dataset targets with the configured class count and document the exact change needed so training completes without the assertion.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100