ROYOLE'S

【代码】json文件的读取和保存。

首先通过以下代码，对问题进行定位with torch.autograd.detect_anomaly(): loss.backward()然后，发现问题出在损失函数上面了：RuntimeError: Function 'CtcLossBackward' returned nan values in its 0th output.检查CTC Loss的参数设置，由于我没有修改原始config中图像初始长宽，而我自己使用的数据集label都比较长，这导致CTC Loss中length比

方式一：直接取布尔值输入：target = torch.Tensor([1,0,0,2,0,0,3])mask = (target > 0)masked_target = target[mask]print(target)print(mask)print(masked_target)输入：target = torch.Tensor([1,0,0,2,0,0,3])mask = target.ge(0)masked_target = torch.masked_sel.

图解PyTorch中的torch.gather函数 - 知乎 (zhihu.com)