skada
skada copied to clipboard
Custom_loss_function
Changed metrics.py to add a new parameter 'criterion' within 'loss_function' with default as 'cross_entropy_loss' and can switch to 'BCELoss'