Feature/class weighted dice and ce loss

DIAGNijmegen / nnUNet

Apache License 2.0

3 stars 14 forks source link

Closed thijsgelton closed 2 years ago

thijsgelton commented 2 years ago

Added trainer class with a weighted CE loss and a SoftDiceLoss (no weighting)
Made all configurable through a command line parameter called --trainer_kwargs (see help text for expected format)
Tested by hand