search

ShivamShrirao / diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
https://huggingface.co/docs/diffusers
Apache License 2.0
1.89k stars 505 forks source link

Question about prior loss #160

Open JyChen9811 opened 1 year ago

JyChen9811 commented 1 year ago

Hello, in my training, the prior loss does not decrease and has been maintained at a certain value. image image How can I solve this problem?

brian6091 commented 1 year ago

What is your prior_loss_weight? Not sure there is a problem per se, as your total loss is decreasing.