custom loss function

[arnon-1]

Hello,

I'd like to try training on images using a custom loss function for training a lora model. In particular, I want to be able to emphasise training on certain parts of images.
I do not request this to be implemented, but I would appreciate a hint of what loss function I should mess with (in the code) for a local quick and dirty solution.

Thanks in advance!

[araleza]

The upcoming 'masked loss' feature should provide what you need:

#589

[arnon-1]

Ah nice, I don't even need to code anything myself :-)