Hey Diego,
I have a couple of questions about your implementation.



 Diego:

F_loss = self.alpha * (1-pt)**self.gamma * BCE_loss



What is alpha for a binary focal loss?  In binary segmentation there is only one class (the one you’re trying to segment)
Your equation isn’t using epsilon, is that intentional?

Thanks!