Add CrossEntropyLoss
and DiceCELoss
wrappers with support for class weights
#787
Job | Run time |
---|---|
4s | |
1m 55s | |
1m 38s | |
8m 16s | |
11m 53s |