Training on custom dataset

[UsamaShaikh1]

Hi, I am trying the code base with a custom dataset with a different number of keypoints. But the model for some reason not learning angles and even after 300 epochs the angle prediction is not even near gt angles. Can you please guide me on this issue. @BrianG13

[BrianG13]

Hi @UsamaShaikh1 , can you add here your losses plot?

What data are you using to train the model? Please note that if you changed the number of keypoints, you need to change the S branch as well to output the correct number of bones and change the FK layer with your body topology