Hello, thank you for sharing good work!
I tried to apply the reg_loss calculated by the step_regularize() function in train_smooth_diffusion.py in other diffusion models.
However, I found out that the reg_loss shakes a lot (1 ~ 100).
What is the expected range of reg_loss?
Or, is there any additional skill necessary to be applied?
Hello, thank you for sharing good work! I tried to apply the reg_loss calculated by the
step_regularize()
function in train_smooth_diffusion.py in other diffusion models. However, I found out that the reg_loss shakes a lot (1 ~ 100). What is the expected range of reg_loss? Or, is there any additional skill necessary to be applied?Thanks:)