RuntimeError: index 24910 is out of bounds for dimension 0 with size 7840

[JinShchen]

shape is: torch.Size([3, 448, 448]) torch.Size([1, 25000]) torch.Size([25000, 3])
Traceback (most recent call last):
File "ctpn_train.py", line 105, in
loss_cls = critetion_cls(out_cls, clss)
File "/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/mnt/lustre/chenjinsheng/project/text_detection/pytorch_ctpn/ctpn_model.py", line 54, in forward
cls_pred = input[0][cls_keep]

RuntimeError: index 24910 is out of bounds for dimension 0 with size 7840

这个index对不上怎么解决呀