训练问题

[Cici0526]

您好，我在复现您代码的时候，出现错误：
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]

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