search

princeton-vl / pose-ae-train

Training code for "Associative Embedding: End-to-End Learning for Joint Detection and Grouping"
BSD 3-Clause "New" or "Revised" License
373 stars 76 forks source link

How to modify the extension ae_loss code to fit Pytorch 1.0? #35

Closed Yishun99 closed 5 years ago

teamo429 commented 5 years ago

@DouYishun I also encountered the same problem... Do you solve it?

Yishun99 commented 5 years ago

@teamo429 https://github.com/DouYishun/loss_collection/blob/master/AEloss/AE_loss.py

FightStone commented 4 years ago

@DouYishun Thank you for providing a python AELOSS. Have you tested it? What's the effect? Looking forward to your reply！