Custom loss function

[DanielSHKao]

I am training my own ControlNet model, and I'd like to add some loss functions to focus on local regions, such as human face or hands. Are there any suggestions on how this can be done?

[mileret]

I think maybe you can modify the p_losses function in LatentDiffusion class.

[bookkkker]

Halo, how do you resolve it? I have the questions toooo.