Hi,
First of all, I'd like to express my appreciation for this work.
While exploring the code, I noticed that there doesn't appear to be an implementation for re-weighting the loss in a manner similar to Focal Loss.
Could you kindly clarify if this functionality is implemented somewhere in the code and I may have missed it? If not, would you consider adding such code in the future?
Thank you again for your great work, and I look forward to your feedback.