Training suddenly terminate after first epoch. Looking for help, plz about fpn.pytorch HOT 5 OPEN

I found that the code runs normally on faster-rcnn. But if I use the code of fpn, it failed. So I guess the problem happens in fpn.py, but I still can't find out why.
What's more, I used this model to train my personal data, if I changed the data back to origin Voc2007, it works. That's strange. I just changed my personal data into the form of Voc2007.
Here is one of my annotation file:

train
VIRAT_S_000000.mp4_0
C:/Users/Kevin Qian/Downloads/images/train/VIRAT_S_000000.mp4_0.jpg

and here is the annotation file in original voc2007

VOC2007
009962.jpg

The VOC2007 Database
PASCAL VOC2007
flickr
246788553


Tool - Wroclaw
Milosz J.


500
375
3

0

chair
Right
1
0

211
192
324
326



person
Unspecified
1
0

162
72
273
248



person
Right
1
0

250
68
473
312



person
Right
1
0

4
2
253
374



diningtable
Unspecified
1
1

358
216
500
375

from fpn.pytorch.

@KevinQian97 I have encountered with the same problem. Have you found out how to solve it?

from fpn.pytorch.

@KevinQian97 @WangTianYuan did you solve this issue?

from fpn.pytorch.

Have you solved the problem? I got the same error.@KevinQian97 @WangTianYuan

from fpn.pytorch.

Have you solved the problem? I got the same error.@KevinQian97 @WangTianYuan

I found that if you use your own dataset to train the model, if it has dirty data, it will cause Nan values in roi_ level in FPN.py. You can try the following modification methods:
roi_ level[roi_ level < 2] = 2
roi_ level[roi_ level > 5] = 5
To
roi_ level[roi_ level < 2] = 2
roi_ level[roi_ level > 5] = 5
roi_ level[roi_ level!=roi_ level]=5

from fpn.pytorch.