How to finetune pretrained mdoel on custom dataset to predict relative transformation of poses?

[ShivaniKamtikar]

Hello, I am confused as to how to use the pretrained model on my own dataset to predict relative transformation of poses.
My dataset contains frames from a video and the corresponding poses. How do I use RAFT for my own training?

Any help is appreciated.

Thank you,
Shivani