pytorch version

[juneleung]

Hello, thanks for your great work.
DeepLabCut has a new pytorch version and I am unfamiliar with tf, so is there any version based on pytorch? Thank you! :)

[Mojackhak]

Hi Juneleung,

From my experience, Anipose is almost fully compatible with the PyTorch version of DeepLabCut, as the high-level API remains unchanged. However, one notable limitation of the current PyTorch version is its lack of support for exporting multiple predictions, even when num_outputs=xx is set in the config.yaml file. This issue prevents the Viterbi filter from functioning properly.

To address this, I made a minor modification to the code. If you're interested, you can try my slightly adjusted version by installing it with:

pip install git+https://github.com/Mojackhak/DeepLabCut.git@pytorch_dlc

I’ve tested this modification on single-animal tracking, and it works as expected.

Best regards!