[Reproducing PhaseNet] Phase Angle not Changing Monotonically

[cryscan]

Hi! I am interesting in PhaseNet and I am re-implementing it on my own. I use joint linear velocities in root frame to train the model as stated in the paper, and I got an MSE loss of ~0.007. But the phase angles I get from the model don't look nice: they swing a lot instead of changing monotonically (video attached). What could possibly go wrong here?

2024-11-21.15-53-45.mp4