skada icon indicating copy to clipboard operation
skada copied to clipboard

Custom_loss_function

Open Som-III opened this issue 10 months ago • 1 comments

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

Som-III avatar Apr 03 '24 16:04 Som-III