Hi,
I’m currently training models for a highly imbalanced medical dataset, so I’m trying to implement the Focal Loss function in order to compensate this.
For my training pipeline, I’m using the train_one_epoch function, and I don´t know how to code this on the function. Thanks in advance 
