Updated for newer version of PyTorch

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

Open dmetehan opened 3 years ago

dmetehan commented 3 years ago

AE loss is implemented in Python and deprecated PyTorch functions are replaced with newer ones.