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

extentions to fit pytorch 0.4.1 #42

Closed notagenius closed 5 years ago

notagenius commented 5 years ago

For the changed API of pytorch. Update /extentions/AE/src/my_lib.c with new_my_lib.c