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

Can the code run on the pytorch 0.4x ? #7

Open moyan007 opened 5 years ago

moyan007 commented 5 years ago

How can I do to make the code suitable for pytorch 0.4+? Anyone can give me some advices? Thank you a lot.