search

mjq11302010044 / RRPN_pytorch

RRPN in pytorch 1.0 ----- Arbitrary-Oriented Scene Text Detection via Rotation Proposals
MIT License
262 stars 57 forks source link

cuda error:device-side assert triggered #29

Closed coorful closed 5 years ago

coorful commented 5 years ago

Sorry to bother you ,i have met a question when i train the rrpn model.

2ced2bb67385a999cf8aae1de049c8f as shown in the figure,it seems that there occurs bug in box_regression.So i print the value of intermediate process variables. I'm surprised to find that the ouput of resnet_layer2 is nan .I try to lower the base lr,but it is useless.

Could you please give me some advice?Thanks a lot. @mjq11302010044