search

bailvwangzi / repulsion_loss_ssd

Repulsion Loss: Detecting Pedestrians in a Crowd. https://arxiv.org/abs/1711.07752
MIT License
234 stars 66 forks source link

The training is easy to collapse #4

Open ShihuaHuang95 opened 6 years ago

ShihuaHuang95 commented 6 years ago

Thanks for your sharing. I have one question, after I have made some changes for running on my environment (Python 3.6, Pytorch 4.1), the training is easy to collapse （ loss be NAN), have you encountered it ? Best,