How to train a smooth manifold of poses?

[HospitableHost]

Hi there,

I'm experiencing an issue about training a smooth manifold.

With your pretrained model, I find that the distances are always zero or very small, which means that your manifold is smooth and your pretrained model can always predict zero distances for plausible poses.

But with my model (which is trained by your codes for training and generating training data), I find that my manifold is not smooth, my model will predict zero distances or small distances for plausible poses, and predict large distances for unplausible poses.

In a word, the variance of distances from my model is larger than yours.

Could you please help me with this problem? I'd really appreciate any assistance you can provide.

Thanks in advance!

[garvita-tiwari]

My model will predict zero distances or small distances for plausible poses, and predict large distances for unplausible poses

Isn't this the desired behavior?
Do you use AMASS_clean, AMASS_noisy and random poses during training?

Can you share a plot of GT distance v/s predicted distance? As long as it's linear, it should be a good model. and we can compare this with the old model.