grasp_diffusion icon indicating copy to clipboard operation
grasp_diffusion copied to clipboard

Loss not decreasing

Open SrinidhiBharadwaj opened this issue 1 year ago • 3 comments

Is it expected to see the loss not go down during training? Loss kind of stagnates after the first epoch which does not make any sense. What am I missing?

SrinidhiBharadwaj avatar Oct 09 '24 22:10 SrinidhiBharadwaj

I encountered the same problem, where the score loss did not decrease, but the sdf loss did decrease, and I have trained for 10,000 steps. I look forward to the author's answer and other ideas.

WuTanKun avatar Apr 14 '25 07:04 WuTanKun

In our implementation, the loss function for the diffusion part was not very trustworthy.

Did you try running the model and visualize the outcome? In our experience, even if the loss was not decreasing much, the generation was good.

robotgradient avatar Apr 16 '25 22:04 robotgradient

Thanks for your response! I have run the model and visualized the results. Although the diffusion loss is not reduced much, the results seem to perform well. I will do more exploration.

WuTanKun avatar Apr 27 '25 06:04 WuTanKun