Hi,
the only way to do some kind ‘sample - weighting’ is by setting reduction='none'
(you’ll get the individual loss values) and multiply them with your custom sample - weight what you should therefore provide. (This is actually the same as what is mentioned in this thread.) But perhaps this is what you’ve already done.
Other than that, I cannot think of an approach.
Regards.