search

liuqidong07 / LEADER-pytorch

[arXiv'24] The official implementation code of LEADER.
https://arxiv.org/abs/2402.02803
MIT License
15 stars 3 forks source link

训练问题 #5

Open Cici0526 opened 4 months ago

Cici0526 commented 4 months ago

您好，我在复现您代码的时候，出现错误： RuntimeError: CUDA error: device-side assert triggered Compile with TORCH_USE_CUDA_DSA to enable device-side assertions. 错误地点是pooled_logits = logits[torch.arange(batch_size, device=logits.device), sequence_lengths] 其中我的batch_size：1； logits.shape是[121,1,131], sequence_lengths:tensor[120] 请问是怎么回事呀

liuqidong07 commented 4 months ago

您好，感谢对我们工作的关注。因为我这两周在赶ddl，然后短时间内也没能复现您说的问题，所以可能需要两周后再来解决这个问题，实在抱歉。祝好