different protein samples

[ZhuLvs]

Dear Developers, thank you for your help!

I would like to ask: how does RF handle the problem of excessively large loss deviations across different timesteps t and different protein samples during training? When I use rigid-body translation MSE loss, I find that the loss fluctuates a lot — ranging from thousands down to less than 1. The overall average loss also seems to oscillate, because it may be stuck in a shaking stage, almost without effective learning. Even when I print the average loss at t < 0.2, I find that it still does not show effective learning.

I look forward to your guidance, thank you! Wish you a pleasant day!