add custom loss arg to supervised_training.py#161

Open

stmartineau99 wants to merge 1 commit intomainfrom

cldice_supervised_training

Contributor

stmartineau99 commented Mar 16, 2026

added custom loss_fn arg to supervised_training
if None, defaults to torch_em.Loss.DiceLoss
custom loss functions are expected to be compatible with ignore_label and mask_channel args (loss wrappers)


          add custom loss arg to supervised_training.py

70b8fcc

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet