Versions for different dependencies

[RitvikKapila]

Bug report and Documentation Ask

Description

The current code doesn't seem to be working with the latest versions of dependencies (pytorch, pytorch_lightning, torchvision, torchaudio, etc.). Please let me know what versions of dependencies is the library designed for.

Current Behavior

I tried running it with the torch 2.1.0 and pytorch-lightning 2.1.0 and it doesn't work and throws a bunch of deprecated warnings and erros.

[pranay-ar]

Hi Ritvik, these are the versions that worked for me with the repo:

pytorch-lightning==1.9.1
torch==1.13.1 torchaudio==0.13.1 torchmetrics==0.11.1 torchvision==0.14.1

Here is the requirements.txt file for other package versions as well. Let me know if this fixes the issue!

[RitvikKapila]

Hey Pranay, thanks a lot for your response. I'll try these versions.

Can you also let me know if you found any pre-trained weights for imu2clip?

[pranay-ar]

I had trained on my own. It should be possible to train with a single GPU like TitanX/M40. There are no publicly released pre-trained weights though.

[RitvikKapila]

Hey Pranay, if it's okay with you, can you please share the pre-trained weights that you're using? Thanks in advance.