Loss computation from samples

[chicodelarosa]

Hi, all.

So, I am working on a custom transform. I want to add a loss term that would compare the samples altered by the transform and those that were not. Up to now, I've seen that loss functions take pred, target, reduction procedure, and other parameters as arguments, but I also want to pass on the samples. Is there an API way of doing this through MMDet?

I will appreciate any help you can provide.